计算图像数据集的RGB均值方差

最新推荐文章于 2022-09-23 16:10:23 发布

Kerixbor

最新推荐文章于 2022-09-23 16:10:23 发布

阅读量2.4k

点赞数

分类专栏： MyTools 文章标签：均值算法深度学习机器学习

本文链接：https://blog.csdn.net/qq_36449923/article/details/121826842

版权

博客内容展示了如何计算图像数据集中RGB通道的均值和标准差，输出结果为B通道均值116.11，标准差32.34；G通道均值108.48，标准差30.98；R通道均值101.99，标准差34.19。

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


import os
import cv2
import numpy as np

path = r'D:\_NLP\_datasets\totaltext\totaltext\training'       #图片保存路径

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)/255
    G_mean = np.mean(per_image_Gmean)/255
    B_mean = np.mean(per_image_Bmean)/255
    stdR = np.std(per_image_Rmean)/255
    stdG = np.std(per_image_Gmean)/255
    stdB = np.std(per_image_Bmean)/255
    return R_mean, G_mean, B_mean, stdR, stdG, stdB

if __name__ == '__main__':
    R, G, B, stdR, stdG, stdB= compute(path)
    print("B=