利用python中的colour_index()函数进行读取图片中的r、g、b值

摆渡人~

已于 2024-03-26 14:36:37 修改

阅读量158

点赞数 3

文章标签： python

于 2024-03-26 14:35:16 首次发布

本文链接：https://blog.csdn.net/qq_52665779/article/details/137045667

版权

该篇文章介绍了如何使用numpy库在Python中定义一个名为colour_index的函数，计算图像数据的色彩特征，如RGB平均值、色差比率等。

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

import numpy as np
def  colour_index(image_data):
    sz1 = image_data.shape
    sz = list(sz1)
    a1 = round(sz[0] / 4)
    a2 = a1 * 3
    b1 = round(sz[1] / 4)
    b2 = b1 * 3
    R = np.mean(np.mean(image_data[a1-1:a2, b1-1: b2,0]))
    G = np.mean(np.mean(image_data[a1-1:a2, b1-1: b2,1]))
    B = np.mean(np.mean(image_data[a1-1:a2, b1-1: b2,2]))
    r2 = R / (R + G + B)
    g2 = G / (R + G + B)
    b2 = B / (R + G + B)
    rb1 = R / B
    rb2 = (R - B) / (R + G + B)
    rb3 = (R + B) / (R + G + B)
    rb4 = (R - B) / (R + B)
    EXR = 1.4 * r2 - g2
    VARI = (g2 - r2) / (g2 + r2 - b2)
    GRVI = (g2 - r2) / (g2 + r2)
    MGRVI = (g2 ** 2 - r2 ** 2) / (g2 ** 2 + r2 ** 2)
    EXG = 2 * g2 - b2 - r2
    GLA = (2 * G - B - R) / (2 * G + B + R)
    index = [R, G, B, r2, g2, b2, rb1, rb2, rb3, rb4, EXR, VARI, GRVI, MGRVI, EXG, GLA]
return index