知识点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)