如何使用opencv和Python在ROI中找到轮廓？

Question

我试图在图像的特定区域找到轮廓。是否可以只显示ROI内部的轮廓而不是图像其余部分的轮廓？我在另一篇类似的帖子中读到我应该使用面具，但我不认为我正确使用它。我是openCV和Python的新手，所以任何帮助都很受欢迎。

import numpy as np
import cv2

cap = cv2.VideoCapture('size4.avi')
x, y, w, h= 150, 50, 400 ,350
roi = (x, y, w, h)

while(True): 
    ret, frame = cap.read()
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    _, thresh = cv2.threshold(gray, 127, 255, 0)
    im2, contours, hierarchy = cv2.findContours(thresh,    cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)

    roi = cv2.rectangle(frame, (x,y), (x+w, y+h), (0,0,255), 2)
    mask = np.zeros(roi.shape,np.uint8)
    cv2.drawContours(mask, contours, -1, (0,255,0), 3)

    cv2.imshow('img', frame)

Answer 1

既然你声称自己是新手，我已经制定了一个解决方案和插图。

请考虑以下内容作为原始图片：

假设以下红色区域是您感兴趣的区域（ROI），您可以在其中找到您的轮廓：

首先，构建相同大小的黑色像素的图像。 必须是相同的尺寸：

black = np.zeros((img.shape[0], img.shape[1], 3), np.uint8) #---black in RGB

现在形成面具并突出显示投资回报率：

black1 = cv2.rectangle(black,(185,13),(407,224),(255, 255, 255), -1)   #---the dimension of the ROI
gray = cv2.cvtColor(black,cv2.COLOR_BGR2GRAY)               #---converting to gray
ret,b_mask = cv2.threshold(gray,127,255, 0)                 #---converting to binary image

现在用原始图像掩饰上面的图像：

fin = cv2.bitwise_and(th,th,mask = mask)

现在使用cv2.findContours()查找上图中的轮廓。

然后使用cv2.drawContours()在原始图像上绘制轮廓。您将最终获得以下内容：

也可能有更好的方法，但这样做是为了让您了解OpenCV中可用的按位AND 操作，该操作专门用于屏蔽 < / p>

Answer 2

要在Python中设置ROI，可以使用标准的NumPy索引such as in this example。

因此，要选择正确的投资回报率，您不能使用cv2.rectangle函数（即绘制矩形），但您可以这样做：

 _, thresh = cv2.threshold(gray, 127, 255, 0)
 roi = thresh[x:(x+w), y:(y+h)]
 im2, contours, hierarchy = cv2.findContours(roi, cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)