Advertisement

Pytorch入门学习——数据加载与预处理的基本方法

阅读量:

torchvision的主要用途。

两种数据集:
1. 所有图像均存储于同一目录下。(此种情形可直接采用 torch.utils.data.DataSet 类进行处理!)
2. 不同类别图像分别存放在独立的文件夹中。(该情况建议使用 torchvision.datasets.ImageFolder(‘image_dir_root’ ) 进行加载。

多数任务所采用的数据形式应属于第一种类型,而第二种通常用于分类任务,例如像imagenet这样的数据集包含1000个类别,每个类别对应一个单独的文件夹。

具体目录结构如下:

复制代码
    root/ants/xxx.png
    root/ants/xxy.jpeg
    root/ants/xxz.png
    .
    .
    .
    root/bees/123.jpg
    root/bees/nsdf3.png
    root/bees/asd932_.png
    
      
      
      
      
      
      
      
      
      
    

注意:

torchvision包的三个主要功能:

  1. 为用户提供了广泛使用的模型,并支持对常见数据集进行直接处理。
  2. 对*

全部评论 (0)

还没有任何评论哟~