pytorch加载自己本地数据集 深度学习数据预处理

数据集存放

将数据集分类存放，不同类别放在不同目录下，目录名即标签，数据集存放格式如下：
root/ants/001.png
root/ants/002.png
root/ants/003.png
…
root/bees/001.png
root/bees/002.png
root/bees/003.png
…

图像数据预处理

1、使用transforms设置图像预处理操作

设置剪裁、缩放、翻转等参数，详细参考链接transforms

# 设置图像数据预处理操作
data_transform = transforms.Compose([
 # 随机缩放剪裁 size 224*224
  transforms.RandomResizedCrop(224),
 # 依概率p水平翻转
  transforms.RandomHorizontalFlip(p=0.5),
 # 转为张量并归一化到[0,1]（是将数据除以255），且会把H*W*C会变成C *H *W
  transforms.ToTensor(),
 # 数据归一化处理
  transforms.Normalize