在最终图像上应用轮廓遮罩

Question

在尝试从背景中分割植物时，我遇到了一个问题 通过色相值创建蒙版并在其上使用close和open运算符后，遇到以下情况：

此后，我想去除图像边缘的小部分，我通过以下操作做到了这一点：

_, cont, heir = cv2.findContours(mask_final, cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE)
contour_sizes = [cv2.contourArea(contour) for contour in cont]
for con, size in zip(cont, contour_sizes):
    if size > 5000:
        mask_final = cv2.drawContours(mask_final, [con], -1, (255, 255, 255), cv2.FILLED)

应用此选项后，斑点已被删除，但是通过以下方式应用：

final = cv2.bitwise_and(img_rgb,img_rgb, mask = mask_final)

我得到以下结果：

可以看出，蒙版没有正确地应用到图像上，有人知道为什么会这样吗？

Answer 1

代替在mask_final函数中使用cv2.drawContours，而是创建一个形状与原始图像相同的新mask并执行以下操作：

mask = np.zeros(img.shape, np.uint8)

for con, size in zip(contours, contour_sizes):
    if size > 5000:
        mask = cv2.drawContours(mask, [con], -1, (255, 255, 255), cv2.FILLED)

final = cv2.bitwise_and(img_rgb, img_rgb, mask = cv2.cvtColor(mask, cv2.COLOR_BGR2GRAY))