文中主要使用两种方式进行彩色图像的颜色增强
主要用到了直方图均衡化原理(代码如下):
import numpy as np
def hist_ave_2(src):
L=np.unique(src)
cdf=(np.histogram(src.flatten(),L.size)[0]/src.size).cumsum()
cdf=(cdf*L.max()+0.5)
return np.interp(src.flatten(),L,cdf).reshape(src.shape)
下面使用RGB模型进行颜色增强
- RGB
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
img=np.array(Image.open("0.jpg"))
img_co=img.copy()
i_r,i_g,i_b=img[:,:,0],img[:,:,1],img[:,:,2]
i_r=hist_ave_2(i_r)
i_g=hist_ave_2(i_g)
i_b=hist_ave_2(i_b)
img_co[:,:,0]=i_r
img_co[:,:,1]=i_g
img_co[:,:,2]=i_b
fig,(ax0,ax1)=plt.subplots(1,2)
fig.dpi=150
ax0.imshow(img)
ax0.set_title("Original")
ax1.imshow(img_co)
ax1.set_title("Color Image Enhancement(RGB)")
- HSI
下面使用模型进行颜色增强
import cv2
img_=cv2.imread("0.jpg")
img_=cv2.cvtColor(img_,cv2.COLOR_BGR2HSV)
#对I通道进行处理
i_i=img_[:,:,2]
i_i=hist_ave_2(i_i)
img_[:,:,2]=i_i
img_=cv2.cvtColor(img_,cv2.COLOR_HSV2RGB)
fig,(ax0,ax1)=plt.subplots(1,2)
fig.dpi=150
ax0.imshow(img)
ax0.set_title("Original")
ax1.imshow(img_)
ax1.set_title("Color Image Enhancement(HSI)")
经过实验HSI模型更适合颜色增强,RGB模型会修改原来的颜色,如有问题,请多多指正