web-dev-qa-db-fra.com

OpenCV - utilisation correcte de cv2.approxPolyDP ()

J'essaie d'extraire une approximation d'un contour dans une image, en utilisant cv2.approxPolyDP(). Voici l'image que j'utilise:

map of UK

Mon code tente d'isoler l'île principale et de définir et tracer l'approximation et la coque du contour. J'ai tracé le contour trouvé en vert, l'approximation en rouge:

import numpy as np
import cv2

# load image and shrink - it's massive
img = cv2.imread('../data/UK.png')
img = cv2.resize(img, None,fx=0.25, fy=0.25, interpolation = cv2.INTER_CUBIC)

# get a blank canvas for drawing contour on and convert img to grayscale
canvas = np.zeros(img.shape, np.uint8)
img2gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)

# filter out small lines between counties
kernel = np.ones((5,5),np.float32)/25
img2gray = cv2.filter2D(img2gray,-1,kernel)

# threshold the image and extract contours
ret,thresh = cv2.threshold(img2gray,250,255,cv2.THRESH_BINARY_INV)
im2,contours,hierarchy = cv2.findContours(thresh, cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE)


# find the main island (biggest area)
cnt = contours[0]
max_area = cv2.contourArea(cnt)

for cont in contours:
    if cv2.contourArea(cont) > max_area:
        cnt = cont
        max_area = cv2.contourArea(cont)

# define main island contour approx. and hull
perimeter = cv2.arcLength(cnt,True)
epsilon = 0.01*cv2.arcLength(cnt,True)
approx = cv2.approxPolyDP(cnt,epsilon,True)

hull = cv2.convexHull(cnt)

# cv2.isContourConvex(cnt)

cv2.drawContours(canvas, cnt, -1, (0, 255, 0), 3)
cv2.drawContours(canvas, approx, -1, (0, 0, 255), 3)
## cv2.drawContours(canvas, hull, -1, (0, 0, 255), 3) # only displays a few points as well.

cv2.imshow("Contour", canvas)
k = cv2.waitKey(0)

if k == 27:         # wait for ESC key to exit
    cv2.destroyAllWindows()

Voici les images résultantes:

enter image description here

La première image trace le contour en vert. La seconde trace l'approximation en rouge - comment puis-je représenter cette approximation comme une courbe fermée continue?

Le documentation n'est pas très clair et le tutoriel non plus, mais ma compréhension est que cv2.approxPolyDP() devrait définir une courbe continue fermée, que je devrais être en mesure de tracer avec cv2.drawContours(). Est-ce exact? Si oui, que fais-je de mal?

pythonopencvcontour
13
Aidenhjj

Le problème ne concerne que la visualisation: drawContours attend un tableau (liste en cas de python) de contours, pas seulement un tableau numpy (qui est renvoyé par approxPolyDP).

La solution est la suivante: remplacement

cv2.drawContours(canvas, approx, -1, (0, 0, 255), 3)

à

cv2.drawContours(canvas, [approx], -1, (0, 0, 255), 3)
23
avtomaton