ImageFolder使用方法

ImageFolder假设所有的文件按文件夹保存好,每个文件夹下面存贮同一类别的图片,文件夹的名字为分类的名字
例如:
载入图片:

train_dataset=ImageFolder('./time_fequence_image/')

此时,看看这个train_dataset的类型:

print(type(train_dataset))

输出结果为:

<class 'torchvision.datasets.folder.ImageFolder'>

查看train_data里面的类别:

print(train_dataset.class_to_idx)

输出就是文件夹的名字和序号:

{'BPSK': 0, 'CW': 1, 'LMF': 2, 'NLMF': 3}

这是一个字典结构的。

print(train_dataset[0][0])
#所有图片的路径和对应的label
print(dataset.imgs)

#没有任何转变,所有返回的还是PIL Image对象
print(dataset[0][1]) #第二维度为1 ,表示label
print(dataset[0][0]) #第二维度为0,表示图片数据

print(dataset.classes)  #根据分的文件夹的名字来确定的类别,就是文件夹的名字,列表结构
print(dataset.class_to_idx) #按顺序为这些类别定义索引为0,1...,字典结构
print(dataset.imgs) #返回从所有文件夹中得到的图片的路径以及其类别,列表结构

猜你喜欢

转载自blog.csdn.net/huatianxue/article/details/108321965