计算图像均值

来源:互联网 发布:淘宝卖水果 编辑:程序博客网 时间:2024/05/29 19:57

计算图像均值是图像预处理中的常见操作,计算机视觉中,深度学习框架输入图片后一般都会进行均值削减,分两种方法,一种是减去均值文件,另一种是生成RGB三个数值。
第一种方法一般参考各自框架的API即可。
第二种方法可以由简单的python代码得到,如下所示:

import osimport cv2from numpy import *#img_dir='/home/zzy/ai_challenger_scene_validation_20170908/scene_validation_20170908/'img_dir='/home/zzy/ai_challenger_scene_train_20170904/scene_train_20170904/'img_list=os.listdir(img_dir)img_size=224sum_r=0sum_g=0sum_b=0count=0for img_name in img_list:    img_path=os.path.join(img_dir,img_name)    img=cv2.imread(img_path)    img=cv2.cvtColor(img,cv2.COLOR_BGR2RGB)    img=cv2.resize(img,(img_size,img_size))    sum_r=sum_r+img[:,:,0].mean()    sum_g=sum_g+img[:,:,1].mean()    sum_b=sum_b+img[:,:,2].mean()    count=count+1sum_r=sum_r/countsum_g=sum_g/countsum_b=sum_b/countimg_mean=[sum_r,sum_g,sum_b]print img_mean
原创粉丝点击