數據集規則化(歸一化):transforms.Normalize()

規則化(Normalize)

pytorch在加載數據時,常需要歸一化,如

transforms.Normalize((0.5,0.5,0.5),(0.5,0.5,0.5))

灰度圖是:

transforms.Normalize(mean=[0.5],std=[0.5])

一般我們加載的數據,RGB的值範圍是[0,255],常見數據集如MNIST的數值範圍是[0,1]。

這裏取0.5是歸一化,把範圍[0,1]的數值調整到[-1,1]。

邊界變化如下:

下界:

0-0.5/0.5 = -1

上界:

1-0.5/0.5 = 1

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章