🔴 실시간 색상 추적기 만들기

특정 색상(빨간색)을 카메라로 실시간 추적하는 간단한 프로그램을 만들어볼게요!

import cv2
import numpy as np

# 웹캠 열기
cap = cv2.VideoCapture(0)

while True:
    # 프레임 읽기
    ret, frame = cap.read()
    if not ret:
        break
        
    # HSV 색상 공간으로 변환 (색상 추적에 더 효과적)
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
    
    # 빨간색 범위 정의 (HSV에서)
    lower_red = np.array([0, 100, 100])
    upper_red = np.array([10, 255, 255])
    
    # 마스크 생성 (빨간색 부분만 흰색, 나머지는 검은색)
    mask = cv2.inRange(hsv, lower_red, upper_red)
    
    # 원본 이미지에 마스크 적용
    result = cv2.bitwise_and(frame, frame, mask=mask)
    
    # 빨간색 객체의 윤곽선 찾기
    contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
    
    # 가장 큰 윤곽선 찾기
    if contours:
        largest_contour = max(contours, key=cv2.contourArea)
        x, y, w, h = cv2.boundingRect(largest_contour)
        
        # 사각형 그리기
        cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)
        cv2.putText(frame, "Red Object", (x, y-10), 
                    cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
    
    # 결과 화면 표시
    cv2.imshow('Original', frame)
    cv2.imshow('Red Detection', result)
    
    # 'q' 키를 누르면 종료
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 자원 해제
cap.release()
cv2.destroyAllWindows()
      

이 코드는 웹캠으로 빨간색 물체를 실시간으로 찾아서 초록색 사각형으로 표시해주는 프로그램이에요. HSV 색상 공간, 마스크 생성, 윤곽선 검출 등 앞서 배운 개념들이 모두 활용되고 있죠! 👍