【预处理】计算RGB通道均值

最新推荐文章于 2023-12-20 16:24:28 发布

行者无疆兮

最新推荐文章于 2023-12-20 16:24:28 发布

阅读量6.3k

点赞数

分类专栏：图像处理 opencv python 文章标签： RGB channels

本文链接：https://blog.csdn.net/qq_30159015/article/details/82117878

版权

python 同时被 3 个专栏收录

108 篇文章 3 订阅

订阅专栏

图像处理

13 篇文章 1 订阅

订阅专栏

opencv

7 篇文章 0 订阅

订阅专栏

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# @Time: 2018/8/16
# @Author: xfLi
# 计算RGB通道均值


import os
import cv2
import numpy as np


path = './datasets/train'
def compute(path):
    file_names = os.listdir(path)
    per_image_Rmean = []
    per_image_Gmean = []
    per_image_Bmean = []
    for file_name in file_names:
        img = cv2.imread(os.path.join(path, file_name), 1)
        per_image_Rmean.append(np.mean(img[:,:,0]))
        per_image_Gmean.append(np.mean(img[:,:,1]))
        per_image_Bmean.append(np.mean(img[:,:,2]))
    R_mean = np.mean(per_image_Rmean)
    G_mean = np.mean(per_image_Gmean)
    B_mean = np.mean(per_image_Bmean)
    return R_mean, G_mean, B_mean

if __name__ == '__main__':
    R, G, B= compute(path)
    print(R, G ,B)