拼凑了一下网上找的代码,实现了批量对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()
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)