[Pytorch] Pytorch torchvision.transforms 函數介紹

1.transforms.ToTensor()

• ToTensor()將shape(H,W,C)轉為shape(C,H,W)的tensor, 將每一個數值歸一化為[0,1], 直接將數值除255

2.transforms,Normalize()

• transforms.Compose([transforms.ToTensor(),transforms.Normalize(std=(0.5,0.5,0.5),mean=(0.5,0.5,0.5))]), 將輸入歸一化至[0,1], 再使用公式 (x-mean)/std, 將圖片的三通到分佈至[-1,1]