transform.ToTensor(),
transform.Normalize(mean=(0.5,0.5,0.5),std=(0.5,0.5,0.5))
那transform.Normalize()是怎么工作的呢?以上面代碼為例,ToTensor()能夠把灰度范圍從0-255變換到0-1之間,而后面的transform.Normalize()則把0-1變換到(-1,1).具體地說,對每個通道而言,Normalize執行以下操作:
image=(image-mean)/std
其中mean和std分別通過(0.5,0.5,0.5)和(0.5,0.5,0.5)進行指定。原來的0-1最小值0則變成(0-0.5)/0.5=-1,而最大值1則變成(1-0.5)/0.5=1.