[python] 파이썬을 사용하여 카메라로부터의 물체 색상 추출하기

14 Dec 2023

python

이번에는 파이썬을 사용하여 OpenCV 라이브러리를 통해 카메라로부터의 물체 색상을 추출하는 방법에 대해 알아보겠습니다.

1. OpenCV 설치하기

먼저, OpenCV를 설치해야 합니다. 아래 명령어를 사용하여 OpenCV를 설치합니다.

pip install opencv-python

2. 카메라로부터의 영상 받아오기

다음으로, 파이썬으로 카메라로부터 영상을 받아오는 코드를 작성합니다. 아래는 OpenCV를 사용하여 카메라로부터 영상을 받아오는 간단한 예제 코드입니다.

import cv2

cap = cv2.VideoCapture(0)

while True:
    ret, frame = cap.read()
    cv2.imshow('frame', frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

위 코드는 카메라로부터 영상을 받아와 화면에 표시하는 기본적인 예제입니다.

3. 색상 추출하기

이제, 받아온 영상으로부터 특정 색상을 추출해보겠습니다. 아래는 빨간색 범위의 색상을 추출하는 코드 예제입니다.

import cv2
import numpy as np

cap = cv2.VideoCapture(0)

while True:
    ret, frame = cap.read()
    
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
    
    lower_red = np.array([0, 100, 100])
    upper_red = np.array([10, 255, 255])
    
    mask = cv2.inRange(hsv, lower_red, upper_red)
    res = cv2.bitwise_and(frame, frame, mask=mask)
    
    cv2.imshow('frame', frame)
    cv2.imshow('mask', mask)
    cv2.imshow('res', res)
    
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

위 코드는 카메라로부터 받아온 영상에서 빨간색의 범위에 해당하는 부분만을 추출하여 보여주는 예제입니다.

마치며

카메라로부터의 영상을 받아와 색상을 추출하는 방법에 대해 알아보았습니다. 이를 응용하여 물체 인식이나 추적 등 다양한 기술들을 구현할 수 있습니다. OpenCV 라이브러리를 잘 활용하여 다양한 컴퓨터 비전 프로젝트에 도전해보세요!