PYTHON实现直方图均衡化

PYTHON实现直方图均衡化

在不调用其他python库函数的情况下完成直方图均衡化的操作并输出图像

代码

// An highlighted block
import cv2
import numpy as np

img = cv2.imread('operationbefore.png')#读图
# 转化为灰度图并获取灰度值直方图列表
# 转为灰色图片
img_gray = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
cv2.imshow("img_gray",img_gray)
cv2.waitKey()
gray_store = [0.0]*256
pr_store = [0]*256
# 获取图像尺寸并记录直方图列表
sp = img_gray.shape
height = sp[0]
width = sp[1]
# print(height, width)
for i in range(height):
    for j in range(width):
        gray_store[img_gray[i][j]] += 1
# 生成频度列表及累加列表
for i in range(len(gray_store)):
    if i == 0:
        pr_store[i] = (gray_store[i] * 255) / (height * width)
    else:
        pr_store[i] = (gray_store[i] * 255) / (height * width) + pr_store[i-1]
# 均衡化操作
for i in range(len(pr_store)):
    k = 0
    while pr_store[i] - k > 0:
        k += 1
    chazhi = k - pr_store[i]
    if chazhi <= 0.5:
        pr_store[i] = k
    else:
        pr_store[i] = k-1
# 根据直方图改变图像
for i in range(height):
    for j in range(width):
        img_gray[i][j] = pr_store[img_gray[i][j]]

# 显示图片
cv2.imshow("img_gray",img_gray)
cv2.waitKey()

相关基础知识链接

https://blog.csdn.net/weixin_41059269/article/details/97028518