[python] OpenCV로 얼굴 인식하기

07 Dec 2023

python

이번에는 Python의 OpenCV 라이브러리를 사용하여 얼굴을 인식하는 방법에 대해 알아보겠습니다. OpenCV는 이미지 및 비디오 처리를 위한 강력한 라이브러리로, 컴퓨터 비전 작업을 위해 널리 사용됩니다. 얼굴 인식은 OpenCV의 주요 기능 중 하나이며, 다양한 응용 프로그램에서 사용될 수 있습니다.

1. OpenCV 설치하기

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

pip install opencv-python

2. 이미지에서 얼굴 인식하기

다음은 이미지에서 얼굴을 인식하는 간단한 예제 코드입니다. 아래 코드를 참고하여 실행해 보세요.

import cv2

# 이미지 파일 읽기
image_path = 'path_to_image.jpg'
image = cv2.imread(image_path)

# Haar cascade 파일 로드
cascade_path = 'haarcascade_frontalface_default.xml'
face_cascade = cv2.CascadeClassifier(cascade_path)

# 얼굴 인식
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
faces = face_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30))

# 얼굴 주변에 사각형 그리기
for (x, y, w, h) in faces:
    cv2.rectangle(image, (x, y), (x+w, y+h), (0, 255, 0), 2)

# 결과 이미지 출력
cv2.imshow('Faces', image)
cv2.waitKey(0)
cv2.destroyAllWindows()

위 코드를 실행하면 이미지에서 인식된 얼굴 주변에 초록색 사각형이 그려진 결과 이미지가 표시됩니다.

3. 비디오에서 얼굴 인식하기

이미지가 아닌 실시간 비디오에서 얼굴을 인식하는 방법도 있습니다. 비디오에서 얼굴을 실시간으로 인식하기 위해서는 다음 코드 예제를 참고하세요.

import cv2

# 비디오 캡처
video_capture = cv2.VideoCapture(0)

# Haar cascade 파일 로드
cascade_path = 'haarcascade_frontalface_default.xml'
face_cascade = cv2.CascadeClassifier(cascade_path)

while True:
    # 비디오 프레임 읽기
    ret, frame = video_capture.read()

    # 얼굴 인식
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    faces = face_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30))

    # 얼굴 주변에 사각형 그리기
    for (x, y, w, h) in faces:
        cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)

    # 결과 비디오 출력
    cv2.imshow('Video', frame)

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

# 비디오 캡처 해제
video_capture.release()
cv2.destroyAllWindows()

위 코드를 실행하면 웹캠 비디오에서 실시간으로 인식된 얼굴 주변에 초록색 사각형이 그려진 결과가 표시됩니다.

마무리

이것은 OpenCV를 사용하여 얼굴을 인식하는 간단한 예제였습니다. OpenCV에는 더 다양한 기능과 응용 프로그램이 있으며, 컴퓨터 비전 작업을 수행할 때 매우 유용한 도구입니다. OpenCV의 자세한 사용 방법에 대해서는 공식 문서를 참조하시기 바랍니다.

위 예제 코드로 여러분도 OpenCV를 통해 얼굴을 인식해 보세요!