在PyTorch中加载图像

我是PyTorch的新手，正在研究GAN模型。我想加载我的图像数据集。使用Keras的方法是：

from keras.preprocessing.image import img_to_array
from keras.preprocessing.image import load_img

def load_images(path, size=(128,128)):
data_list = list()
# enumerate filenames in directory, assume all are images
for filename in listdir(path):
    # load and resize the image
    pixels = load_img(path + filename, target_size=size)
    # convert to numpy array
    pixels = img_to_array(pixels)
    # store. 
    data_list.append(pixels)
return asarray(data_list)
# dataset path
path = 'mypath/'
# load dataset A
dataA = load_images(path + 'A/')
dataAB = load_images(path + 'B/')

我想知道如何在PyTorch中做同样的事情。 感谢您的帮助。谢谢

导入torchvision，torch
从torchvision导入数据集、模型和转换
def加载训练（根路径、目录、批量大小、kwargs）：
transform=transforms.Compose(
[transforms.Resize（[256，256]），
transforms.RandomCrop（224），
transforms.RandomHorizontalFlip（），
transforms.ToTensor（））
data=datasets.ImageFolder（root=root\u path+dir，transform=transform）
train\u loader=torch.utils.data.DataLoader（数据，批量大小=批量大小，随机播放=真，最后一次播放=真，**kwargs）
回程列车装载机

---

我希望它能工作…

您可以随时使用OpenCV之类的库来加载图像。Pytorch有一个名为torchvision的子库，其中包含大量教程。