彩色图像的直方图均衡使用python opencv给出错误

Question

错误：合并中的断言失败 (0

在合并函数中

cv2.merge(channels,img2)

如果参数被替换为：

cv2.merge(img2,channels)

它不会出错，但均衡前后的直方图是一样的。我可以在这段代码中做什么。

代码：

import cv2,cv
import cv2.cv as cv
import numpy as np
from matplotlib import pyplot as plt

capture = cv.CaptureFromCAM(0)
img = cv.QueryFrame(capture)
img_size = cv.GetSize(img)

width,height = img_size
size = width,height,3
channels = np.zeros(size , np.uint8) 


while (1):
img = cv.QueryFrame(capture)
img = np.asarray(img[:,:])

cv2.imshow("original",img)
hist = cv2.calcHist([img],[2],None,[256],[0,256])

#convert img to YCR_CB
img2 = cv2.cvtColor(img,cv2.COLOR_BGR2YCR_CB)

#split image to Y, CR, CB
cv2.split(img2,channels)

#histogram equalization to Y-MATRIX
cv2.equalizeHist(channels[0],channels[0])

#merge this matrix to reconstruct our colored image
cv2.merge(channels,img2)

#convert this output image to rgb
rgb = cv2.cvtColor(img2,cv2.COLOR_YCR_CB2BGR)
hist2 = cv2.calcHist([rgb],[2],None,[256],[0,256])
plt.plot(hist)
plt.plot(hist2)
plt.show()

Answer 1

不要使用 split 和 merge，而是利用 numpy 切片。

img2[:, :, 0] = cv2.equalizeHist(img2[:, :, 0]) 
# or run a small loop over each channel

Answer 2

你在这里弄错了 split() 。它返回频道。

由于您没有捕获返回值，因此您的频道未初始化

>>> import cv2
>>> help(cv2.split)
Help on built-in function split in module cv2:

split(...)
    split(m[, mv]) -> mv

所以它应该看起来像：

channels = cv2.split(img2)

请避免旧的 cv api，而是始终坚持使用 cv2。 （使用 cv2.VideoCapture，而不是 cv.CaptureFromCAM）