[python] PyQt를 활용한 얼굴 인식 애플리케이션 개발

30 Nov 2023

python

얼굴 인식 기술은 현대의 다양한 응용 프로그램에서 사용되고 있습니다. 이 기술은 보안 시스템, 사진 인식 애플리케이션, 게임 및 사회적으로 대화하는 로봇 등에 사용될 수 있습니다. PyQt는 파이썬에서 사용되는 GUI 프레임워크이며, 이를 활용하여 얼굴 인식 애플리케이션을 개발하는 방법을 알아보겠습니다.

필요한 라이브러리 설치

가장 먼저, PyQt를 설치해야 합니다. PyQt는 아래의 명령을 통해 설치할 수 있습니다.

pip install PyQt5

다음으로, 얼굴 인식을 위해 OpenCV 라이브러리도 설치해야 합니다. 아래 명령을 통해 설치할 수 있습니다.

pip install opencv-python

얼굴 인식 애플리케이션 구현하기

import sys
import cv2
from PyQt5.QtGui import QImage, QPixmap
from PyQt5.QtWidgets import QApplication, QDialog, QLabel, QVBoxLayout

class FaceDetectionApp(QDialog):
    def __init__(self):
        super().__init__()

        # UI 요소 초기화
        self.label = QLabel(self)
        self.layout = QVBoxLayout()
        self.layout.addWidget(self.label)
        self.setLayout(self.layout)

        # 얼굴 인식을 위한 OpenCV 초기화
        self.face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
        self.video_capture = cv2.VideoCapture(0)

    def detect_faces(self):
        while True:
            ret, frame = self.video_capture.read()

            # 이미지를 회색으로 변환하여 얼굴 인식 수행
            gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
            faces = self.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)

            # 프레임을 PyQt에 표시하기 위해 이미지 변환
            image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
            image = QImage(image.data, image.shape[1], image.shape[0], QImage.Format_RGB888)

            # PyQt 라벨에 이미지 표시
            self.label.setPixmap(QPixmap.fromImage(image))
            self.label.setScaledContents(True)

    def closeEvent(self, event):
        self.video_capture.release()

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = FaceDetectionApp()
    window.detect_faces()
    window.show()
    sys.exit(app.exec_())

위의 코드는 PyQt를 사용하여 얼굴 인식 애플리케이션을 개발하는 예입니다. FaceDetectionApp 클래스는 PyQt의 QDialog 클래스를 상속받고 얼굴 인식을 위해 OpenCV를 사용합니다.

detect_faces 메서드는 카메라로부터 이미지를 읽고, 그레이스케일로 변환한 후 얼굴을 인식하고 사각형을 그립니다. 그리고 PyQt의 라벨 위젯에 인식된 이미지를 표시합니다.

실행하기

위의 코드를 실행하면 웹캠으로부터 입력을 받아 얼굴을 인식하고 화면에 표시하는 얼굴 인식 애플리케이션이 실행됩니다.

본 예제에서는 OpenCV의 haarcascade_frontalface_default.xml 파일을 사용하여 얼굴을 인식합니다. 필요에 따라 다른 얼굴 인식 모델을 사용할 수 있습니다.

얼굴 인식 애플리케이션을 개발하는데 활용할만한 PyQt와 OpenCV의 다양한 기능에 대한 자세한 내용은 공식 문서 및 OpenCV 공식 사이트를 참조하시기 바랍니다.