Opencv Python 2.7 中的轮廓检测

Question

我一直在尝试识别 Open Cv (python 2.7) 中红色物体的轮廓，我们已经成功识别它们。但是，我想检测红色物体（左或右）的位置，但我没有成功。如果有人能给我这样做的代码或步骤，我将非常感激。

我们目前识别红色物体的代码如下：

import numpy as np
import cv2
cap = cv2.VideoCapture(0)
while(1):
  _, frame = cap.read()
  hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
  lower_color = np.array([0, 50, 50])
  upper_color = np.array([60, 255, 255])
  mask = cv2.inRange(hsv, lower_color, upper_color)
  mask = cv2.erode(mask, None, iterations=2)
  mask = cv2.dilate(mask, None, iterations=2)
  res = cv2.bitwise_and(frame, frame, mask=mask)

  cnts = cv2.findContours(mask.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)[-2]
  cv2.drawContours(frame, cnts, 0, (127, 255, 0), 3)
 print cnts
 cv2.imshow('frame', frame)
 cv2.imshow('mask', mask)
 cv2.imshow('res', res)
 cv2.imshow('contours', frame)



 k = cv2.waitKey(5) & 0xFF
 if k == 27:
    print "release"

    break
 cap.release()
 cv2.destroyAllWindows()

Answer 1

嗯，你离获得这个职位只有一步之遥。您可以在轮廓周围创建一个boundingrect，然后您可以计算它的中心以获得对象的坐标。

你也可以试试minEnclosingCircle 这会给你它的中心和半径。这可能会更直接地找到中心:)

Here你可以找到一个关于这两个函数的小教程，但是在 c++ 中。

在 python 中会是这样的

...  
cnts = cv2.findContours(mask.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)[-2]
cv2.drawContours(frame, cnts, 0, (127, 255, 0), 3)
(x,y),radius = cv2.minEnclosingCircle(cnts[0])
center = (int(x),int(y))
radius = int(radius)
cv2.circle(frame, center, radius, (255, 0, 0), 3)
...

在这种情况下，中心将是您的对象的位置。此代码仅考虑数组中的第一个轮廓...