python-3.x - 在 OpenCV Python 中屏蔽特定区域

Question

我有一个奶牛场的图像。在图像中，有两个感兴趣区域 (ROI)。在每个投资返回率中，我希望一切都是黑色的。

ROIs各角的坐标为-

1= [0, 1440]
2= [0, 1087]
3= [977, 80]
4= [1925, 67]
5= [2560, 800]
6= [2560, 1440]
7= [1465, 1440]
8= [1455,60]

我正在使用以下代码来屏蔽红色区域并使 ROI 中的所有内容变黑。

import cv2, numpy as np
original_frame = cv2.imread("original.jpg")
frame = original_frame.copy()

# pts - location of the corners of the roi
pts = np.array([[0, 1450], [0, 1087], [977, 80], [1925, 67], [2560, 800], [2560, 1440]])
(x,y,w,h) = cv2.boundingRect(pts)

pts = pts - pts.min(axis=0)
mask = np.zeros(original_frame.shape, np.uint8)
cv2.drawContours(mask, [pts], -1, (255, 255, 255), -1, cv2.LINE_AA)
result = cv2.bitwise_and(original_frame, mask)

cv2.imwrite("out.jpg", result)

结果很好，但仍然覆盖了顶部的一些额外区域。

如果我试图通过改变

pts = np.array([[1455,60], [1925, 67], [2560, 800], [2560, 1440],[1465, 1440] ])

我得到了一个完全错误的结果——

有什么方法可以获得蓝色 ROI 的正确结果吗？

Answer 1

我让整个过程变得复杂。有一个非常直接和简单的答案。

import numpy as np
import cv2
import matplotlib.pyplot as plt

# create a polygons using all outer corners of the ROI
external_poly = np.array( [[[1458,1440],[0,1440],[0,0],[2560,0],[2560,740], [1940,60], [1453,60]]], dtype=np.int32 )
im = cv2.imread("original.png", 1)
cv2.fillPoly( im , external_poly, (0,0,0) )
cv2.imwrite("output.jpg", im) 

此方法创建一个多边形，将 ROI 之外的所有内容都带走，并用黑色填充多边形。