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包的三个主要功能:
- 为用户提供了广泛使用的模型,并支持对常见数据集进行直接处理。
- 对*
全部评论 (0)
还没有任何评论哟~
