파이썬과 OpenCV를 사용한 동영상에서 눈동자 추적하기

17 Oct 2023

opencv

이 블로그 포스트에서는 파이썬과 OpenCV를 사용하여 동영상에서 눈동자를 추적하는 방법을 다룰 것입니다.

OpenCV란?

OpenCV(Open Source Computer Vision Library)는 오픈 소스 컴퓨터 비전 라이브러리로, 이미지 및 영상 처리에 사용됩니다. 파이썬에서도 OpenCV를 사용하여 영상 처리 작업을 수행할 수 있습니다.

동영상에서 눈동자 추적하기

필요한 라이브러리 설치하기: 우리는 cv2(OpenCV)와 numpy 라이브러리를 사용할 것입니다. 이 라이브러리들을 설치하기 위해 다음 명령어를 이용하세요.

pip install opencv-python
pip install numpy

동영상 파일 불러오기: 우리는 눈동자를 추적할 동영상 파일을 불러와야 합니다. 다음 코드를 실행하여 동영상 파일을 불러오세요.

import cv2

# 동영상 파일 경로
video_path = 'path_to_video_file.mp4'

# 동영상 파일 불러오기
cap = cv2.VideoCapture(video_path)

눈동자 추적하기: 눈동자를 추적하기 위해 Haar Cascade 분류기를 사용할 수 있습니다. 다음 코드를 사용하여 눈동자를 추적하세요.

# Haar Cascade 분류기 경로
eye_cascade_path = 'haarcascade_eye.xml'

# Haar Cascade 분류기 불러오기
eye_cascade = cv2.CascadeClassifier(eye_cascade_path)

while True:
    # 프레임 읽어오기
    ret, frame = cap.read()
    
    if not ret:
        break
    
    # 그레이스케일로 변환
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    
    # 눈 영역 탐지
    eyes = eye_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30))
    
    for (x, y, w, h) in eyes:
        # 눈 영역에 사각형 그리기
        cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)
        
    # 동영상 재생 및 결과 출력
    cv2.imshow('Video', frame)
    
    if cv2.waitKey(1) == ord('q'):
        break

# 종료
cap.release()
cv2.destroyAllWindows()

결론

이번 블로그 포스트에서는 파이썬과 OpenCV를 사용하여 동영상에서 눈동자를 추적하는 방법을 알아보았습니다. OpenCV는 다양한 컴퓨터 비전 작업에 유용하게 사용될 수 있으며, 동영상 처리 작업에도 많이 활용됩니다.

목차

OpenCV란?

동영상에서 눈동자 추적하기

결론

참고 자료