心得体会
接上一节的Dataset,本节对直接加载的原始数据进行预处理操作。所有的Transforms均可通过map
方法传入,实现对指定数据列的处理。
Common Transforms
一系列通用Transforms,如Compose
composed = transforms.Compose(
[
vision.Rescale(1.0 / 255.0, 0),
vision.Normalize(mean=(0.1307,), std=(0.3081,)),
vision.HWC2CHW()
]
)
Vision Transforms
Rescale
变换用于调整图像像素值的大小,包括两个参数:
- rescale:缩放因子。
- shift:平移因子。
Normalize
变换用于对输入图像的归一化,包括三个参数:
- mean:图像每个通道的均值。
- std:图像每个通道的标准差。
- is_hwc:bool值,输入图像的格式。True为(height, width, channel),False为(channel, height, width)。
HWC2CHW
变换用于转换图像格式。在不同的硬件设备中可能会对(height, width, channel)或(channel, height, width)两种不同格式有针对性优化
在图像处理方面,上述数据变换能够满足深度学习的基本要求,并以此建立训练的基础。