Как нарисовать контур загороженного / перекрывающегося объекта? - PullRequest
Новая
       34

Как нарисовать контур загороженного / перекрывающегося объекта?

1 голос
/ 11 февраля 2020

Я пытаюсь заставить OpenCV распознавать, что красная область - это один контур, а не сегментировать его на два отдельных контура. Я пытался возиться с иерархией, но безуспешно. Любые идеи?

Изображение и код приведены ниже. 

import cv2
import numpy as np
black = (0,0,0)
# All of these are in hsv format
lb=np.array([0,181,0])
ub=np.array([6,255,255])
# Load img
img = cv2.imread('test.jpg')
img=cv2.resize(img,(400,400))

kernelOpen=np.ones((5,5))
kernelClose=np.ones((20,20))
#convert BGR to HSV
imgHSV= cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
# create the Mask
mask = cv2.inRange(imgHSV, lb, ub)
maskso=cv2.morphologyEx(mask,cv2.MORPH_OPEN,kernelOpen)
masksc=cv2.morphologyEx(maskso,cv2.MORPH_CLOSE,kernelClose)
conts,hiers=cv2.findContours(masksc.copy(),cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE)
# Create convex hull
hull = []
for i in range(len(conts)):
    # creating convex hull object for each contour
    hull.append(cv2.convexHull(conts[i], False))
# Draw contour
cv2.drawContours(img,hull,-1,black,2)

cv2.imshow("image", img)
cv2.waitKey(5000)
cv2.destroyAllWindows()

Without contours drawn

With contours drawn

1 Ответ

3 голосов
/ 11 февраля 2020

Замена этого фрагмента кода 

# Create convex hull
hull = []
for i in range(len(conts)):
    # creating convex hull object for each contour
    hull.append(cv2.convexHull(conts[i], False))
# Draw contour
cv2.drawContours(img,hull,-1,black,2)

на 

# Create convex hull
hull = np.zeros((0, 1, 2), np.int32)
for i in range(len(conts)):
    hull = np.concatenate((hull, conts[i]), axis=0)
hull = cv2.convexHull(hull, False)
# Draw contour
cv2.drawContours(img,[hull],-1,black,2)

Я получаю следующий вывод, который, как я полагаю, вам нужен:

Output

Вместо того, чтобы собирать отдельные выпуклые оболочки каждого контура, я собираю все контуры, а затем получаю выпуклую оболочку всех точек контура.

Надеюсь, это поможет! 

----------------------------------------
System information
----------------------------------------
Platform:    Windows-10-10.0.16299-SP0
Python:      3.8.1
NumPy:       1.18.1
OpenCV:      4.2.0
----------------------------------------
...