一、Transforms是什么?
torchvision中对数据(图片)进行预处理的函数。
二、使用方法
1.使用示例
代码如下(示例):
transform = transforms.Compose(
[transforms.Resize((32, 32)),
transforms.ToTensor(),
transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])
(1):resize():改变数据的大小
(2):totensor():转成张量形式
(3):Normolize():进行标准化处理,即减去均值再除以标准差()
例如:(0 - 0.5)/0.5=-1,(1-0.5)/0.5=1,将数据归一化至(-1,1)。
原因:数据如果分布在(0,1)之间,可能实际的bias,就是神经网络的输入b会比较大,而模型初始化时b=0的,这样会导致神经网络收敛比较慢,经过Normalize后,可以加快模型的收敛速度。