Pytorch 读取指定文件夹中的图片

1. 文件夹与源代码的位置关系

2. 代码 (0.py)

import matplotlib.pyplot as plt
import torch
import torchvision
from torchvision import datasets, transforms

data_transfrom = transforms.Compose([  # 对读取的图片进行以下指定操作
    transforms.Resize((300, 300)),     # 图片放缩为 (300, 300), 统一处理的图像最好设置为统一的大小,这样之后的处理也更加方便
    transforms.ToTensor(),             # 向量化,向量化时 每个点的像素值会除以255,整个向量中的元素值都在0-1之间      
])

img = datasets.ImageFolder('.', transform=data_transfrom)  # 指明读取的文件夹和读取方式,注意指明的是到文件夹的路径,不是到图片的路径

imgLoader = torch.utils.data.DataLoader(img, batch_size=2, shuffle=False, num_workers=1)  # 指定读取配置信息

inputs, _ = next(iter(imgLoader))
print(_[0].item())           # 打印返回的值
inputs = inputs / 2 + 0.5    # 使得整个图像的像素值都在0.5之上，使得处理后的图像偏白
inputs = torchvision.utils.make_grid(inputs)   # 降维,因为操作之前的inputs是4维的, (1, 3, w, h) 3代表通道数, w,h代表图片的宽高
inputs = inputs.numpy().transpose((1, 2, 0))   # transpose((1, 2, 0)) 是将(3, w, h) 变为 (w ,h, 3), 因为这种格式才是图像存储的标准格式
plt.imshow(inputs)           # 展示,这里会一块展示batch_size张图片,因为它们是一块被读出来的
plt.show()

3. 运行结果