#?-*-?coding:?utf-8?-*-
“““
Created?on?Mon?Apr?22?22:02:51?2019

@author:?MSIK
“““

import?os
import?numpy?as?np
import?cv2
from?tqdm?import?tqdm
?
#?calculate?means?and?std
filepath?=?‘D:/SemanticSegmentation/pytorch-semseg/datasets/VOC/VOC2019/JPEGImages‘
pathDir?=?os.listdir（filepath）


CNum?=?1000?????#?挑選多少圖片進(jìn)行計(jì)算

#random.randint（lowhighsize）
filenameArray?=?np.random.randint（1len（pathDir）CNum）
?
img_h?img_w?=?256?256
imgs?=?np.zeros（[img_w?img_h?3?1]）
means?stdevs?=?[]?[]
?
#for?idx?in?tqdm（range（len（pathDir）））:
for?idx?in?tqdm（range（CNum））:
????filename?=?pathDir[idx]
????img_path?=?os.path.join（filepath?filename）
?
????img?=?cv2.imread（img_path）
????img?=?cv2.resize（img?（img_h?img_w））
????img?=?img[:?:?:?np.newaxis]
????
????imgs?=?np.concatenate（（imgs?img）?axis=3）
#?????????print（i）
?
imgs?=?imgs.astype（np.float32）/255.
?
?
for?i?in?tqdm（range（3））:
????pixels?=?imgs[::i:].ravel（）
????means.append（np.mean（pixels））
????stdevs.append（np.std（pixels））
?
#?cv2?讀取的圖像格式為BGR，PIL/Skimage讀取到的都是RGB不用轉(zhuǎn)
means.reverse（）?#?BGR?-->?RGB
stdevs.reverse（）
?
print（“normMean?=?{}“.format（means））
print（“normStd?=?{}“.format（stdevs））
print（‘transforms.Normalize（normMean?=?{}?normStd?=?{}）‘.format（means?stdevs））