python图像处理：直方图均衡化

最新推荐文章于 2024-05-09 19:24:51 发布

WUNNAN

最新推荐文章于 2024-05-09 19:24:51 发布

阅读量1.5k

点赞数

分类专栏： Python 文章标签： python 图像处理开发语言

本文链接：https://blog.csdn.net/weixin_51154479/article/details/128161550

版权

Python 专栏收录该内容

14 篇文章 7 订阅

订阅专栏

理论参考链接：【图像处理算法】直方图均衡化

算法实现

# 对单通道的处理函数
def channel_equalize(img):
    od = img.reshape(-1)
    # 对像素计数
    tmp = {}
    for num in od:
        tmp[num] = tmp.get(num, 0)+1
    tmp_list = list(tmp.items())
    tmp_list = sorted(tmp_list, key=lambda x:x[0], reverse=False)
    n = {}
    for data in tmp_list:
        n[data[0]] = data[1]

    # 像素分布频率
    for key, _ in n.items():
        n[key] /= len(od)

    # 像素累计分布频率
    pre = 0
    for key, _ in n.items():
        pre += n[key]
        n[key] = pre

    # 均衡化
    for key, _ in n.items():
        n[key] = int(n[key]*(len(n)-1)+1)
    for i in range(img.shape[0]):
        for j in range(img.shape[1]):
            img[i, j] = n[img[i, j]]
    return img