pytorch 之 imagefloder的用法

问题的关键在于如何确定图片和类别的对应关系

from torchvision.datasets import ImageFolder

dataset = ImageFolder('F:/my_code/data/0dB/train/')

读入数据

print(dataset.imgs[2])

此时，dataset.imgs是一个列表，里面有很多元素，选中第2个
输出的是

('F:/my_code/data/0dB/train/BFSK\\BFSK100.jpg', 0)

也就是说，imagefolder应该是以文件夹名字的顺序进行排列的
BFSK对应的0
BPSK对应的1
等等……