DataLoader以及图片处理操作———Pytorch
一、DataLoader
torch.utils.data.DataLoader(
# 数据集;批大小;是否乱序
dataset,batch_size=1,shuffle=False,
# 读取数据的进程数
num_workers=0,
# 当样本数不能被batchsize整除时,
# 是否舍弃最后一批数据
drop_last=False)
二、图片处理
# 组合图片的操作
train_transform = transforms.Compose([
transforms.Resize((32,32)), # 缩放
transforms.RandomCrop(32, padding=4), # 裁剪,属于数据扩增,在训练时使用、测试时不用
transforms.ToTensor(), # 图片转为张量,同时归一化像素值从[0,255]到[0,1]
transforms.Normalize(norm_mean,norm_std), # 标准化
])
train_transform(img)
版权声明:本文为qq_25841513原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。