[파이썬] opencv-python 카메라에서의 물체 거리 추정

07 Sep 2023

opencv python

카메라에서 물체의 거리를 추정하는 것은 컴퓨터 비전에서 중요한 작업입니다. OpenCV-Python은 카메라에서 물체를 탐지하고 추정하는 데 매우 유용한 도구입니다.

이 글에서는 Python을 사용하여 OpenCV-Python을 통해 카메라에서 물체의 거리를 추정하는 방법을 알아보겠습니다.

1. OpenCV 설치하기

먼저, OpenCV-Python을 설치해야 합니다. 다음 명령어를 사용하여 설치할 수 있습니다:

pip install opencv-python

2. 카메라에서 비디오 스트림 읽기

OpenCV-Python을 사용하여 카메라에서 비디오 스트림을 읽을 수 있습니다. 다음 코드를 사용하여 카메라로부터 영상을 읽어오세요.

import cv2

cap = cv2.VideoCapture(0)  # 카메라 장치 번호 (0부터 시작)

while True:
    ret, frame = cap.read()  # 영상 읽기
    cv2.imshow("Camera Stream", frame)  # 읽어온 영상 출력

    # 'q' 키를 누르면 종료
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()  # 카메라 종료
cv2.destroyAllWindows()  # 창 닫기

3. 물체의 거리 추정하기

카메라로부터 영상을 읽어오는 것이 성공했다면, 이제 물체의 거리를 추정할 차례입니다. 물체의 크기와 카메라의 내부 파라미터를 알고 있어야 합니다. 다음 코드는 물체 거리 추정을 위한 예제입니다.

import cv2

# 카메라 캘리브레이션 (카메라 내부 파라미터 설정)
# ...

# 물체 크기 (단위: cm)
object_width = 10

cap = cv2.VideoCapture(0)  # 카메라 장치 번호 (0부터 시작)

while True:
    ret, frame = cap.read()  # 영상 읽기
    cv2.imshow("Camera Stream", frame)  # 읽어온 영상 출력

    # 물체의 크기와 거리에 따른 관심 영역 설정
    # ...

    # 물체의 크기를 측정하여 픽셀 단위에서 실제 거리로 변환
    object_pixel_width =  # 물체의 픽셀 단위 크기 측정
    object_distance = (object_width * focal_length) / object_pixel_width  # 거리 추정식

    # 물체 거리 출력
    print("물체 거리:", object_distance, "cm")

    # 'q' 키를 누르면 종료
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()  # 카메라 종료
cv2.destroyAllWindows()  # 창 닫기

결론

OpenCV-Python을 사용하여 카메라에서 물체의 거리를 추정해보았습니다. 먼저, OpenCV를 설치하고 카메라로부터 영상을 읽어오는 방법을 알아보았습니다. 그리고 물체의 크기와 카메라 내부 파라미터를 설정한 후, 물체 거리를 추정하는 방법을 살펴보았습니다. 이제 당신은 OpenCV-Python을 사용하여 카메라에서 물체의 거리를 측정할 수 있습니다!