一、transforms运行机制
1.1 torchvision——计算机视觉工具包
- torchvision.transforms : 常用的图像预处理方法
- torchvision.datasets : 常用数据集的dataset实现, MNIST, CIFAR-10, ImageNet等
- torchvision.model: 常用的模型预训练, AlexNet, VGG, ResNet, GoogLeNet等
1.2 torchvision.transforms : 常用的图像预处理方法
- 数据中心化, 数据标准化
- 缩放, 裁剪, 旋转, 翻转, 填充, 噪声添加
- 灰度变换, 线性变换, 仿射变换
- 亮度、饱和度及对比度变换
示例:
norm_mean = [0.485, 0.456, 0.406] # 设置标准化的均值
norm_std = [0.229, 0.224, 0.225]