老卫带你学---深度学习计算图像均值

老卫带你学—深度学习计算图像均值

在我们做深度学习研究的时候，如果可以在图像预处理的时候加入去均值的操作，那么可以提高我们训练的速度以及精准度。以下是老卫写的通过opencv来求出图像的均值。

# -*- coding:utf-8 -*-
#作者：卫毅然
#联系邮箱：2205492446@qq.com
import os
import cv2
from numpy import *


img_dir='/home/hades/software/face-py-faster-rcnn-master/data/VOCdevkit2007/VOC2007/JPEGImages'
img_list=os.listdir(img_dir)
img_size=1024
sum_r=0
sum_g=0
sum_b=0
count=0

for img_name in img_list:
    print img_name
    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+1

sum_r=sum_r/count
sum_g=sum_g/count
sum_b=sum_b/count
img_mean=[sum_r,sum_g,sum_b]
print img_mean

然后我们将其均值放入我们的config文件中，就可以开始下一步训练。