批量对RGB图片进行直方图均衡化处理

拼凑了一下网上找的代码，实现了批量对RGB图片进行直方图均衡化处理，也不知道对不对，希望各位提下意见

python代码：

#encoding:utf-8
import cv2
import numpy as np
import matplotlib.pyplot as plt
import os

file_dir = r'./concrete_4/1/' #需要批量处理图片的文件夹
save_dir = './concrete_4/output/'#保存处理后图片的文件夹

for img_name in os.listdir(file_dir):
    img_path = file_dir + img_name
    img = cv2.imread(img_path, cv2.IMREAD_GRAYSCALE)#以灰度图方式读取RGB图片
    print('正在处理',img_name)
    equ = cv2.equalizeHist(img)#将读取的灰度图进行直方图均衡化处理
    cv2.imwrite(save_dir + img_name[0:-4] + '.jpg', equ)#保存的图片格式
    
    #这里显示每张图片的处理前的直方图和处理后的直方图
    # plt.hist(img.ravel(), 256)
    # plt.figure()
    # plt.hist(equ.ravel(), 256)
    # plt.show()
    # cv2.waitKey(0)
    # cv2.destroyAllWindows()