PyTorch DataLoader Images
通过使用PyTorch中的torchvision模块,可以较为方便的构建dataset和dataloader
首先确保你的数据是按照这种方式放置
dataset/classA/1.png
dataset/classA/2.png
dataset/classB/1.png
dataset/classB/2.png
import torch
from torchvision import transforms, datasets, utils
data_transform = transform
原创
2020-12-03 17:30:17 ·
252 阅读 ·
0 评论