计算图像均值

最新推荐文章于 2022-12-15 11:20:45 发布

Never-guess

最新推荐文章于 2022-12-15 11:20:45 发布

阅读量1.4w

点赞数 4

CC 4.0 BY-SA版权

分类专栏：计算机视觉文章标签： python 计算机视觉深度学习图像预处理均值

本文链接：https://blog.youkuaiyun.com/qq_20791919/article/details/78241095

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

import os
import cv2
from 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=224
sum_r=0
sum_g=0
sum_b=0
count=0

for 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+im