celebA数据集的mean和std计算结果
- mean: (0.50612009, 0.42543493, 0.38282761)
- std: (0.26589054, 0.24521921, 0.24127836)
另附计算代码
import numpy as np
import cv2
import os
path = './img_align_celeba'
means = [0, 0, 0]
stdevs = [0, 0, 0]
index = 1
num_imgs = 0
img_names = os.listdir(path)
for img_name in img_names:
num_imgs += 1
print(img_name)
img = cv2.imread(os.path.join(path, img_name))
img = np.asarray(img)
img = img.astype(np.float32) / 255.
for i in range(3):
means[i] += img[:, :, i].mean()
stdevs[i] += img[:, :, i].std()
print(num_imgs)
means.reverse()
stdevs.reverse()
means = np.asarray(means) / num_imgs
stdevs = np.asarray(stdevs) / num_imgs
print("normMean = {}".format(means))
print("normStd = {}".format(stdevs))
print('transforms.Normalize(normMean = {}, normStd = {})'.format(means, stdevs))