0. 往期内容
[二]深度学习Pytorch-张量的操作:拼接、切分、索引和变换
[七]深度学习Pytorch-DataLoader与Dataset(含人民币二分类实战)
[八]深度学习Pytorch-图像预处理transforms
[九]深度学习Pytorch-transforms图像增强(剪裁、翻转、旋转)
[十]深度学习Pytorch-transforms图像操作及自定义方法
深度学习Pytorch-图像预处理transforms
1. torchvision
2. transforms运行机制
3. transforms.Normalize(mean, std, inplace = False)
(1)功能:逐个channel
对图像进行标准化(均值为0
,标准差为1
),output=(input-mean)/std
;
(2)参数:
mean:
各个通道的均值,list
;
std:
各个通道的标准差,list
;
inplace:
是否进行原地操作;
(3)注意:
标准化操作可以加快模型的收敛;
(4)代码示例:
norm_mean = [0.485, 0.456, 0.406]
norm_std = [0.229, 0.224, 0.225]
transforms.Normalize(norm_mean, norm_std)