pytorch 入门Load Dataset

知识点1、读取数据
知识点2、显示第一个图像信息
知识点3、图像转tensor
知识点4、转batch
知识点5、以label排序

from torchvision.datasets import ImageFolder
import matplotlib.pyplot as plt

知识点1
这里的文件夹是包含整个数据的文件夹，文件的文件夹，再下面才是具体的图片

folder_set = ImageFolder('./data/testload')
print(folder_set.class_to_idx)
print(folder_set.imgs)

知识点2
显示第一个图像的信息

im, label = folder_set[0]
plt.imshow(im)
plt.show()
print(label)

知识点3
把图像转化为tensor

from torchvision import transforms as tfs
data_tf =  tfs.ToTensor()
folder_set = ImageFolder('./data/testload', transform=data_tf)
im, label = folder_set[0]
print(im.shape)

知识点4
把图像集转化为batch的形式

from torch.utils.data import DataLoader
train_data1 = DataLoader(folder_set, batch_size=3, shuffle=True)
for im, label in train_data1:
    print(label)
# 显示第一个数据
im, laebel = next(iter(train_data1))

知识点5
以第二列为标准排序

batch.sort(key=lambda x: len(x[1]), reverse=True)