레슨 10: 색상 인식 | MechDog 파이썬 학습

🎨

MechDog 색상 인식

카메라 센서로 색상을 감지하고 LED로 반응하기

🎯 학습 목표

• 카메라 센서로 색상 인식하기
• 색상별 LED 제어 프로그래밍
• 실제 MechDog 하드웨어 코드 분석
• 색상 기반 인터랙션 시스템 구현

🔧 실제 MechDog 색상 인식 코드

# main.py - 실제 MechDog 색상 인식 코드
import Hiwonder
import time
import Hiwonder_IIC
from HW_MechDog import MechDog

# 객체 초기화
iic2 = Hiwonder_IIC.IIC(2)
cam = Hiwonder_IIC.ESP32S3Cam(iic2)  # 카메라
i2c1 = Hiwonder_IIC.IIC(1)
i2csonar = Hiwonder_IIC.I2CSonar(i2c1)  # LED 제어용
mechdog = MechDog()

time.sleep(5)

while True:
    color_list = cam.color_recognition()  # 색상 인식
    print(color_list)
    
    if cam.RED in color_list:  # 빨간색 감지
        i2csonar.setRGB(0, 250, 0, 0)  # 빨간색 LED
    
    if cam.GREEN in color_list:  # 초록색 감지
        i2csonar.setRGB(0, 0, 250, 0)  # 초록색 LED
    
    if cam.BLUE in color_list:  # 파란색 감지
        i2csonar.setRGB(0, 0, 0, 250)  # 파란색 LED
    
    time.sleep(0.1)

💻 색상 인식 시뮬레이션

# MechDog 색상 인식 시뮬레이션
import random

class ColorRecognition:
    def __init__(self):
        self.RED = "RED"
        self.GREEN = "GREEN" 
        self.BLUE = "BLUE"
        self.colors = [self.RED, self.GREEN, self.BLUE]
        
    def color_recognition(self):
        """랜덤 색상 감지 시뮬레이션"""
        detected = []
        for color in self.colors:
            if random.random() < 0.3:  # 30% 확률
                detected.append(color)
        return detected

class MechDogColorSystem:
    def __init__(self):
        self.cam = ColorRecognition()
        self.current_led = "꺼짐"
        
    def setRGB(self, light, r, g, b):
        """LED 색상 설정"""
        if r > 0 and g == 0 and b == 0:
            self.current_led = "빨간색"
        elif r == 0 and g > 0 and b == 0:
            self.current_led = "초록색"
        elif r == 0 and g == 0 and b > 0:
            self.current_led = "파란색"
        else:
            self.current_led = "꺼짐"
        print(f"💡 LED: {self.current_led}")
    
    def run_detection(self):
        """색상 감지 실행"""
        for i in range(10):
            print(f"\n--- 감지 {i+1} ---")
            colors = self.cam.color_recognition()
            print(f"🎨 감지된 색상: {colors}")
            
            if self.cam.RED in colors:
                self.setRGB(0, 250, 0, 0)
                print("🔴 빨간색 반응!")
                
            elif self.cam.GREEN in colors:
                self.setRGB(0, 0, 250, 0)
                print("🟢 초록색 반응!")
                
            elif self.cam.BLUE in colors:
                self.setRGB(0, 0, 0, 250)
                print("🔵 파란색 반응!")
                
            else:
                print("⚪ 인식된 색상 없음")

# 시스템 실행
color_system = MechDogColorSystem()
print("🤖 MechDog 색상 인식 시스템 시작!")
color_system.run_detection()
print("\n🎉 색상 인식 테스트 완료!")

색상 인식 코드를 실행해보세요!

📝 정리

✅ 핵심 내용

• 카메라 센서: color_recognition()으로 색상 감지
• 색상 상수: cam.RED, cam.GREEN, cam.BLUE
• LED 제어: setRGB(light, r, g, b)
• 실시간 처리: while 반복문으로 지속 감지