ABOUT ME

-

Today
-
Yesterday
-
Total
-
  • OpenCV Python 강좌 - 도형 검출하기(Shape Detection)
    OpenCV/OpenCV Python 강좌 2019. 2. 13. 08:36

    이미지에서 도형을 검출하는 방법을 설명합니다.






    유튜브 영상에서 사용한 코드와 테스트 이미지입니다.


    import cv2 as cv

    def setLabel(image, str, contour):
       (text_width, text_height), baseline = cv.getTextSize(str, cv.FONT_HERSHEY_SIMPLEX, 0.7, 1)
       x,y,width,height = cv.boundingRect(contour)
       pt_x = x+int((width-text_width)/2)
       pt_y = y+int((height + text_height)/2)
       cv.rectangle(image, (pt_x, pt_y+baseline), (pt_x+text_width, pt_y-text_height), (200,200,200), cv.FILLED)
       cv.putText(image, str, (pt_x, pt_y), cv.FONT_HERSHEY_SIMPLEX, 0.7, (0,0,0), 1, 8)


    img_color = cv.imread('test.png', cv.IMREAD_COLOR)
    cv.imshow('result', img_color)
    cv.waitKey(0)

    img_gray = cv.cvtColor(img_color, cv.COLOR_BGR2GRAY)
    cv.imshow('result', img_gray)
    cv.waitKey(0)

    ret,img_binary = cv.threshold(img_gray, 127, 255, cv.THRESH_BINARY_INV|cv.THRESH_OTSU)
    cv.imshow('result', img_binary)
    cv.waitKey(0)

    contours, hierarchy = cv.findContours(img_binary, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_SIMPLE)

    for cnt in contours:
       size = len(cnt)
       print(size)

       epsilon = 0.005 * cv.arcLength(cnt, True)
       approx = cv.approxPolyDP(cnt, epsilon, True)

       size = len(approx)
       print(size)

       cv.line(img_color, tuple(approx[0][0]), tuple(approx[size-1][0]), (0, 255, 0), 3)
       for k in range(size-1):
           cv.line(img_color, tuple(approx[k][0]), tuple(approx[k+1][0]), (0, 255, 0), 3)

       if cv.isContourConvex(approx):
           if size == 3:
               setLabel(img_color, "triangle", cnt)
           elif size == 4:
               setLabel(img_color, "rectangle", cnt)
           elif size == 5:
               setLabel(img_color, "pentagon", cnt)
           elif size == 6:
               setLabel(img_color, "hexagon", cnt)
           elif size == 8:
               setLabel(img_color, "octagon", cnt)
           elif size == 10:
               setLabel(img_color, "decagon", cnt)
           else:
               setLabel(img_color, str(size), cnt)
       else:
           setLabel(img_color, str(size), cnt)

    cv.imshow('result', img_color)
    cv.waitKey(0)



    포스트 작성시에는 문제 없었지만 이후 문제가 생길 수 있습니다.
    댓글로 알려주시면 빠른 시일내에 답변을 드리겠습니다.

    여러분의 응원으로 좋은 컨텐츠가 만들어집니다. 지금 본 내용이 도움이 되었다면 유튜브 구독 부탁드립니다. 감사합니다 : )

    유튜브 구독하기


    댓글 0

Designed by Tistory.