【2020更新】python 获取数据集的means和stdevs(均值、方差)
·
import numpy as np
from tqdm import tqdm
from glob import glob
import cv2
import random
import os
means = [0, 0, 0]
stdevs = [0, 0, 0]
index = 1
num_imgs = 0
imgslist = glob('/users/xxx/xxx/*.jpg')
for imgpath in tqdm(imgslist):
num_imgs += 1
img = cv2.imread(imgpath)
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()
means = np.asarray(means) / num_imgs
stdevs = np.asarray(stdevs) / num_imgs
print('normMean = {} normStd = {}'.format(means,stdevs))
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)