[python] PySimpleGUI에서 사물 감지 및 추적 기능 사용하기

30 Nov 2023

python

이번 글에서는 PySimpleGUI를 사용하여 사물 감지 및 추적 기능을 구현하는 방법에 대해 알아보겠습니다.

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

먼저, 필요한 라이브러리를 설치해야합니다. 아래 명령어를 사용하여 OpenCV와 PySimpleGUI를 설치하세요.

pip install opencv-python
pip install PySimpleGUI

사물 감지하기

감지할 사물 이미지를 불러와서 OpenCV의 CascadeClassifier를 사용하여 사물을 감지할 수 있습니다. 아래는 간단한 예제입니다.

import cv2

# 카메라 또는 이미지 파일을 불러옵니다.
image = cv2.imread("object_detection_image.jpg")

# 학습된 분류기 모델을 로드합니다.
cascade = cv2.CascadeClassifier("haarcascade_frontalface_default.xml")

# 이미지를 그레이스케일로 변환합니다.
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# 사물을 감지합니다.
faces = 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("Object Detection", image)
cv2.waitKey(0)
cv2.destroyAllWindows()

사물 추적하기

감지된 사물을 추적하기 위해 몇 가지 추가적인 단계가 필요합니다. 아래는 사물 감지 및 추적을 함께 수행하는 예제입니다.

import cv2

# 카메라 또는 이미지 파일을 불러옵니다.
image = cv2.imread("object_tracking_image.jpg")

# 학습된 분류기 모델을 로드합니다.
cascade = cv2.CascadeClassifier("haarcascade_frontalface_default.xml")

# 이미지를 그레이스케일로 변환합니다.
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# 사물을 감지합니다.
faces = cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30))

# 감지된 사물을 추적합니다.
for (x, y, w, h) in faces:
    # 사각형의 중심 좌표를 계산합니다.
    x_center = int(x + w / 2)
    y_center = int(y + h / 2)
    
    # 사각형의 중심 좌표를 표시합니다.
    cv2.circle(image, (x_center, y_center), 5, (0, 255, 0), -1)
    
    # 사각형 주위에 사각형을 그립니다.
    cv2.rectangle(image, (x, y), (x+w, y+h), (0, 255, 0), 2)

# 결과 이미지를 표시합니다.
cv2.imshow("Object Tracking", image)
cv2.waitKey(0)
cv2.destroyAllWindows()

실행 결과

위의 코드를 실행하면, 사물이 감지 및 추적되는 결과 이미지가 표시됩니다.

결론

이번 글에서는 PySimpleGUI를 사용하여 사물 감지 및 추적 기능을 구현하는 방법을 알아보았습니다. 이를 응용하여 다양한 사물 감지 및 추적 프로그램을 개발할 수 있습니다.

참고 자료

```