OpenCV - usando cv2.approxPolyDP () correctamente
Estoy tratando de extraer una aproximación de un contorno en una imagen, usandocv2.approxPolyDP()
. Aquí está la imagen que estoy usando:
Mi código intenta aislar la isla principal y definir y trazar la aproximación del contorno y el casco del contorno. He trazado el contorno encontrado en verde, la aproximación en rojo:
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()
Aquí están las imágenes resultantes:
La primera imagen traza el contorno en verde. El segundo traza la aproximación en rojo: ¿cómo trazo esta aproximación como una curva cerrada continua?
losdocumentación no es terriblemente claro y tampoco lo es eltutorial, pero entiendo quecv2.approxPolyDP()
debería definir una curva cerrada continua, con la que debería poder trazarcv2.drawContours()
. ¿Es eso correcto? Si es así, ¿qué estoy haciendo mal?