[python] 파이썬을 사용하여 카메라로부터 특정 색상 인식하기

14 Dec 2023

python

우선, OpenCV(Open Source Computer Vision Library)를 사용하여 카메라로부터 비디오를 캡처하고 특정 색상을 인식할 수 있습니다. 다음은 파이썬 코드의 예시입니다.

import cv2
import numpy as np

# 카메라로 비디오를 캡처합니다
cap = cv2.VideoCapture(0)

while True:
    # 프레임을 읽어옵니다
    _, frame = cap.read()

    # BGR에서 HSV 색상 공간으로 변환합니다
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

    # 특정 색 범위를 정의합니다(파란색 예시)
    lower_blue = np.array([110,50,50])
    upper_blue = np.array([130,255,255])

    # HSV 이미지에서 파란색 영역을 찾습니다
    mask = cv2.inRange(hsv, lower_blue, upper_blue)

    # 마스킹된 이미지에서 파란색 영역을 추출합니다
    res = cv2.bitwise_and(frame,frame, mask= mask)

    # 원본 비디오와 마스킹된 이미지를 보여줍니다
    cv2.imshow('Original',frame)
    cv2.imshow('Mask',mask)
    cv2.imshow('Result',res)

    # 'q' 키를 누르면 종료합니다
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 캡처를 해제하고 윈도우를 닫습니다
cap.release()
cv2.destroyAllWindows()

위 코드는 파이썬에서 OpenCV 라이브러리를 사용하여 카메라로부터 비디오를 캡처하고, HSV 색상 공간으로 변환한 후, 특정 색상 범위를 정의하여 해당 색상을 마스킹하고 추출하는 방법을 보여줍니다.

더 자세한 내용은 OpenCV의 공식 문서를 참고하시기 바랍니다.

참고 문헌:

OpenCV 공식 문서: https://docs.opencv.org/