이미지 Dataset에 대한 mean, std 구하기, 정규화

사진 이미지는 이미지 각각 명도, 채도가 다르기 때문에 학습하기 전에 학습의 성능을 높이기 위해 이미지 픽셀에 대한 평균(mean)과 표준편차(std)를 구하여 전처리 작업을 해준다.

CIFAR10 data에 대해서 실습해보았다.

import torch
import torchvision
import torchvision.transforms as transforms

transform = transforms.Compose([transforms.ToTensor()])

train_data = torchvision.datasets.CIFAR10(root='./data', train=True, download=True, transform=transform)
print(train_data.data.shape)

mean = train_data.data.mean(axis=(0, 1, 2))
std = train_data.data.std(axis=(0, 1, 2))

print(mean/255)
print(std/255)

result :

reference :
http://www.gisdeveloper.co.kr/?p=8168