天天看點

torchvision中Transform的normalize參數含義

比如原來的tensor是三個次元的,值在0到1之間,那麼經過以下變換之後就到了-1到1區間。 

transforms.Normalize((.5,.5,.5),(.5,.5,.5))
           

原本以為第一個參數應該是(0,0,0)才是歸一化到均值為0。但是通過transforms的源碼發現:

``input[channel] = (input[channel] - mean[channel]) / std[channel]``
           

也就是說((0,1)-0.5)/0.5=(-1,1)。

即參數為變換過程,而非最終結果。

繼續閱讀