Line detection, findContours(), drawContours()

✣

Background

앞선 포스트에서 검출한 에지 맵에서 에지 화소는 1, 에지가 아닌 화소는 0으로 표시되었다. 사람 눈에는 에지가 연결된 선분으로 보이는데, 에지 맵에는 연결 관계가 암시적으로 나타나 있을 뿐 명시적으로 표현되어 있지 않다. 

이들을 연결하여 경계선으로 변환하고 경계선을 직선으로 변환하면 이후 단계인 물체 표현이나 인식에 무척 유리하다. 

 

경계선 찾기

OpenCV의 findContours 함수를 활용하면 에지 맵에서 경계선을 찾아 연속된 점들의 집합을 리스트로 표현할 수 있다. 

findCountours 함수에 경계선을 찾을 에지 영상을 첫번째 인수에 입력하면, 경계선을 찾아서 객체에 저장을 해준다. 

 

또 OpenCV의 drawContours 함수는 영상에 경계선을 그린다. 첫 번째 인수는 경계선을 그려 넣을 영상이고, 두 번째 인수는 경계선을 저장한 객체이다. 세 번째 인수를 -1로 설정하면 모든 경계선을 그려주며 양수로 설정하면 해당 번호에 해당하는 경계선 하나만 그려준다. 네 번째 인수는 색과 두께를 지정한다. 

import cv2 as cv
import numpy as np

img = cv.imread('light.jpg')
gray = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
canny = cv.Canny(gray, 30,100)

contour, hierarchy = cv.findContours(canny, cv.RETR_LIST, cv.CHAIN_APPROX_NONE)
lcontour = []
for i in range(len(contour)):
    if contour[i].shape[0] > 100:
        lcontour.append(contour[i])
cv.drawContours(img, lcontour, -1, (0, 255, 0), 3)
cv.imshow('Original with contours', img)
cv.imshow('Canny', canny)
cv.waitKey()
cv.destroyAllWindows()