0

我是 Python 新手。我找到了一个在线计算图像直方图的代码。我想计算图像局部区域的直方图，所以我尝试使用蒙版。这是我的代码：

 i_rgb1 = cv2.imread(im1)
   img1 = cv2.cvtColor(i_rgb1, cv.CV_BGR2HSV)
   hist2 = np.zeros(img2.shape)

bins = np.arange(256).reshape(256, 1)
color = [(255, 0, 0), (0, 255, 0), (0, 0, 255)]

for labelx in xrange(len(label)):
    temp = labels_map.copy()
    temp[labels_map != label[labelx]] = 0
    temp[labels_map == label[labelx]] = 255
    cv2.imwrite('mask.png', temp)
    for ch, col in enumerate(color):
        hist_item1 = cv2.calcHist([img1], [ch], temp, [256], [0, 255])

其中 labels_map 是一个图像矩阵，由每个像素的标签分配组成。但是当我运行这段代码时，我收到一条错误消息

OpenCV Error: Assertion failed (!mask.data || mask.type() == CV_8UC1) in unknown function

请帮我解决这个错误。

4

1 回答 1

1

我认为这是因为 temp 不是uint8掩码数组，您可以对其进行转换：

cv2.calcHist([img1], [ch], temp.astype(np.uint8), [256], [0, 255])

或者，当您创建它时：

temp = (labels_map == label[labelx]).astype(np.uint8)
于 2013-03-18T12:26:01.677 回答