相关文章推荐
睿智的甜瓜  ·  pandas ...·  3 周前    · 
讲道义的毛豆  ·  vue3 ...·  2 周前    · 
高大的消防车  ·  vue ...·  2 周前    · 
重感情的夕阳  ·  warinning: ...·  11 月前    · 
俊逸的青蛙  ·  jquery mobile ...·  1 年前    · 
还单身的充电器  ·  jupyter ...·  1 年前    · 
首发于 Python学习
完整版Pytorch训练图像分类--数据增强/标准化

完整版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