OpenCV/OpenCV 강좌

두 직선 사이의 각도 구하는 OpenCV Python 예제

webnautes 2023. 10. 18. 22:29
반응형

두 직선 사이의 각도를 구하는 OpenCV Python 예제 코드입니다.

 

2023.2.25  최초작성



두 개의 직선을 그리면 사이 각을 출력해줍니다. 

 

왼쪽 마우스 버튼을 클릭하여 첫번째 직선의 출발점을 지정합니다. 

 

마우스 왼쪽 버튼을 클릭하여 첫번째 직선의 끝점을 지정해주면 직선이 그려집니다. 

 

마우스 왼쪽 버튼 클릭으로 두번째 직선의 시작점을 지정해줍니다. 

 

마우스 왼쪽 버튼 클릭으로 두번째 직선의 끝점을 지정해주면 직선이 그려집니다. 

 

터미널에 두 직선의 사이각이 출력됩니다. 직선으로 생성되는 두 개의 각도 중 작은 각도가 출력됩니다. 



스페이스바 키를 누르면 화면이 지워집니다. 이제 다시 직선을 그려서 각도를 다시 출력해볼 수 있습니다. 



전체 코드입니다. 

 

import cv2
import numpy as np


def mouse_event(event, x, y, flags, param):
    global count,pre_x, pre_y
   
    if event == cv2.EVENT_FLAG_LBUTTON:

        count = count + 1

        if count % 2 == 0:
            cv2.circle(img, (x,y), 10, (255,0,0), -1)
            cv2.line(img, (pre_x, pre_y), (x, y), (255, 255, 255), 2)  

            list_vector.append(np.array([x-pre_x, y-pre_y]))

            if len(list_vector) == 2:
                radian = np.arccos(np.dot(list_vector[0], list_vector[1])/(np.linalg.norm(list_vector[0])*np.linalg.norm(list_vector[1])))
                theta = radian * 180/np.pi

                theta = np.fmin(theta, 180.0-theta)

                print(theta)
                count = 0
                list_vector.clear()
               
        else:
            pre_x = x
            pre_y = y
            cv2.circle(img, (x,y), 10, (0,0,255), -1)


img_empty = np.zeros( (500,500,3), dtype=np.uint8)
img = img_empty.copy()

count = 0
pre_x = -1
pre_y = -1
list_vector = []

while True:

    text = "Clear Screen: space bar"
    cv2.putText(img, text, (20,50), cv2.FONT_HERSHEY_SIMPLEX, 1, (255,255,255), 1 )


    cv2.imshow('set location', img)
    cv2.setMouseCallback("set location", mouse_event, img)
   
    key = cv2.waitKey(1)
    if key == 27: # esc key
        break
    elif key == 32: #spacebar key
        img = img_empty.copy()




참고

 https://darkpgmr.tistory.com/121 

 

반응형