[python] 파이썬을 사용하여 동영상에서의 사람 인식하기

14 Dec 2023

python

사람 인식 기술은 컴퓨터 비전 분야에서 매우 중요한 부분입니다. 파이썬을 사용하여 동영상에서의 사람을 식별하는 방법을 살펴보겠습니다.

필요한 라이브러리 설치하기

먼저, OpenCV와 imutils 라이브러리를 설치해야 합니다. 다음 명령을 사용하여 설치할 수 있습니다.

pip install opencv-python
pip install imutils

동영상에서의 사람 인식하기

다음은 OpenCV를 사용하여 동영상에서의 사람을 인식하는 간단한 예제 코드입니다.

import cv2
import imutils

# 동영상 파일 열기
video_file = 'sample_video.mp4'
cap = cv2.VideoCapture(video_file)

# 사람 인식을 위한 모델 불러오기
hog = cv2.HOGDescriptor()
hog.setSVMDetector(cv2.HOGDescriptor_getDefaultPeopleDetector())

# 동영상에서 프레임별로 반복
while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break

    # 사람을 인식할 수 있도록 동영상 크기 조정
    frame = imutils.resize(frame, width=min(800, frame.shape[1]))

    # 사람을 인식하고 사각형으로 표시
    (rects, weights) = hog.detectMultiScale(frame, winStride=(4, 4), padding=(8, 8), scale=1.05)
    for (x, y, w, h) in rects:
        cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)

    # 화면에 표시
    cv2.imshow('Video', frame)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

위 코드는 OpenCV에서 제공하는 HOGDescriptor를 사용하여 동영상에서 사람을 인식하고 사각형으로 표시합니다.

이 예제를 실행하면 동영상에서 인식된 사람을 식별할 수 있습니다.

결론

파이썬을 사용하여 동영상에서의 사람을 인식하는 방법을 알아보았습니다. OpenCV와 imutils를 활용하여 사람 인식 기술을 구현할 수 있습니다. 이러한 기술은 보안 시스템, 동영상 분석 등 다양한 분야에서 활용될 수 있습니다.