Python实现计算图像RGB均值

最新推荐文章于 2024-07-05 03:23:29 发布

蹦跶的小羊羔

最新推荐文章于 2024-07-05 03:23:29 发布

阅读量2w

点赞数 8

分类专栏： python 文章标签： python RGB均值

本文链接：https://blog.csdn.net/yql_617540298/article/details/83617512

版权

该博客介绍如何使用Python遍历文件夹，计算每张图片的RGB平均值，并展示整体RGB均值。文章指出，由于cv2.imread()读取顺序为BGR，需注意颜色通道顺序。同时提醒读者，避免路径中出现中文以防止读取错误。

摘要由CSDN通过智能技术生成

要求

存在一个文件夹内有若干张图像，需要计算每张图片的RGB均值，并计算全部图像的RGB均值。

代码

# -*- coding: utf-8 -*-
"""
Created on Thu Nov  1 10:43:29 2018

@author: Administrator
"""

import os
import cv2
import numpy as np
 
 
path = 'C:/Users/Administrator/Desktop/rgb'
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_Bmean.append(np.mean(img[:,:,0]))
        per_image_Gmean.append(np.mean(img[:,:,1]))
        per_image_Rmean.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)