python - 如何使用掩码计算直方图 OPENCV？

Question

我只需要在图像的一部分上计算直方图，但该部分具有圆形(如圆盘)。我创建蒙版以在图像上找到该部分

    cv2.rectangle(mask,(0, 0), (width, height), (0,0,0), -1)
    cv2.circle(mask,(int(avgkrug[0]),int(avgkrug[1])),radijusp2,(255,255,255),-1)
    cv2.circle(mask,(int(avgkrug[0]),int(avgkrug[1])),radijusp1,(0,0,0),-1)

使用上面的代码，我找到了我感兴趣的“圆盘形”区域。 现在我正在尝试计算直方图:

 for ch, col in enumerate(color):
    hist_item = cv2.calcHist([img],[ch],mask,[256],[0,255])
    ...

但是出现了这个错误

 error: (-215) !mask.data || mask.type() == CV_8UC1 in function cv::calcHist

但是，如果我将掩码保存在 dics 上并使用 cv2.imread() 读取它，则不会出现此错误。 我也试过使用这条线

  hist_item = cv2.calcHist([slika],[ch],mask.astype(np.uint8),[256],[0,255])

如何使用我创建的掩码来计算直方图，这样我就不需要从光盘上 w/r 了？

Answer 1

您创建的掩码需要是 uint8 类型，因此在创建掩码时将其设为 uint8，然后将其传递给计算直方图。

mask = np.zeros(image.shape[:2], dtype="uint8")

现在通过传递原始图像和相应的掩码来计算直方图。

hist_item = cv2.calcHist([image],[ch],mask,[256],[0,255])