OpenCV：图像直方图均衡equalizeHist

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接： https://blog.csdn.net/u013419318/article/details/102544734

import cv2
import matplotlib.pyplot as plt
import numpy as np
import os

def histogram_equalization(source_path,out_path):
    if not os.path.exists(source_path):
        return
    if not os.path.exists(out_path):
        os.makedirs(out_path)
    for img in os.listdir(source_path):
        img_path = os.path.join(source_path,img)
        image = plt.imread(img_path)
        # gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
        print(img_path)
        # 彩色图像均衡化,需要分解通道 对每一个通道均衡化

        # (b, g, r) = cv2.split(image)
        # bH = cv2.equalizeHist(b)
        # gH = cv2.equalizeHist(g)
        # rH = cv2.equalizeHist(r)
        # # 合并每一个通道
        # result = cv2.merge((bH, gH, rH))
        result = cv2.equalizeHist(image)

        cv2.imwrite(os.path.join(out_path,img),result)

if __name__ == '__main__':
    histogram_equalization('H:\\CVLabel\\0901~0910\\20190211-210304 {7884C098-91F1-4D59-9459-4DB9BE1C7336}(SHDT-0901(B))\\ZXB_LC01D\\orgImg',
                           'H:\\CVLabel1\\0901~0910\\20190211-210304 {7884C098-91F1-4D59-9459-4DB9BE1C7336}(SHDT-0901(B))\\ZXB_LC01D\\orgImg')