pytorch 准备自己数据库

构建图片数据库一般有两种方法：

1、所有图片放在一个文件夹内，另外有一个txt文件显示标签。

2、不同类别的图片放在不同的文件夹内，文件夹就是图片的类别。

方法2会用到 ImageFolder

ImageFolder 是一个通用的数据加载器，数据集中的数据以以下方式组织

root/dog/xxx.png

root/dog/xxy.png

root/dog/xxz.png


root/cat/123.png

root/cat/nsdf3.png

root/cat/asd932_.png

datasets.ImageFolder(root="root folder path", [transform, target_transform])

使用时要注意图片的存储格式，如上所示

用此函数进行处理的时候，会自动会图片的label命名 0，1，3... 方便接下来的loss计算

class_names = image_datasets['train'].classes 可以会获得cat、dog 等组成的列表

 

参考链接（内有详细代码哦）：

用python将cifar-10二进制数据集转成图片格式《菜鸟教程》   https://blog.csdn.net/guohuifengby/article/details/62424299

pytorch学习：准备自己的图片数据-布布扣-bubuko.com          http://www.bubuko.com/infodetail-2304938.html

Pytroch 自己的人脸数据+迁移学习（resnet18） - CSDN博客  https://blog.csdn.net/sinat_37787331/article/details/80149034