完整版Pytorch训练图像分类--数据增强/标准化
计算机视觉博士在读
公众号 AI末班车：完整版Pytorch训练图像分类--数据篇
公众号 AI末班车：完整版Pytorch训练图像分类--Dataset篇
实际应用过程中，我们会在数据进入模型之前进行一些预处理。Pytorch中这些数据增强方法主要放在了transforms.py文件中。
# 数据预处理设置
normMean = [0.4948052, 0.48568845, 0.44682974]# 第一行
normStd = [0.24580306, 0.24236229, 0.2603115] #第二行
normTransform = transforms.Normalize(normMean, normStd) # 第三行 归一化操作
trainTransform = transforms.Compose([
transforms.Resize(32),
transforms.RandomCrop(32, padding=4),
transforms.ToTensor(),
normTransform