YOLOv8로 로봇 눈 만들기, 물체 인식 30분이면 끝

로봇에게 눈이 있다면 세상이 어떻게 보일까요? 카메라를 달면 되잖아 — 라고 생각하기 쉬운데, 카메라로 "보는" 것과 "이해하는" 것은 완전히 다른 이야기예요. 오늘은 YOLOv8을 사용해서 로봇이 물체를 실시간으로 인식하게 만드는 방법을 알려드릴게요. 진짜 30분이면 돼요!

👁️ YOLO가 뭔데 이렇게 유명할까?

YOLO는 "You Only Look Once"의 약자예요. 이미지를 한 번만 봐도 물체의 종류와 위치를 동시에 파악하는 딥러닝 모델이에요. 마치 사람이 방에 들어가면 한눈에 "의자, 테이블, 컵이 있네"라고 파악하는 것처럼요.

YOLOv8은 Ultralytics에서 개발한 최신 버전으로, 속도와 정확도 모두에서 이전 버전을 크게 앞서요. 특히 로봇용으로 최적화하면 일반 웹캠으로도 초당 30프레임 이상의 실시간 물체 인식이 가능해요. 이 정도면 로봇이 주변을 실시간으로 인식하며 행동하기에 충분하죠.

📌 Step 1: 설치 — 진짜 한 줄이면 끝

설치가 이렇게 쉬운 AI 도구는 처음이었어요.

pip install ultralytics

끝이에요. 진짜 이게 다예요! 필요한 의존성 패키지까지 전부 자동으로 설치돼요. PyTorch도 없으면 알아서 깔아줘요. 이전 YOLO 버전들은 설치가 복잡해서 환경 설정에만 반나절 걸렸는데, v8은 진짜 혁명이에요.

📌 Step 2: 기본 물체 인식 테스트

설치가 됐으면 바로 테스트해볼까요? 파이썬에서 단 3줄이면 이미지 속 물체를 인식할 수 있어요. YOLO 모델을 로드하고, 이미지를 넘기면 끝이에요.

기본 제공되는 모델은 COCO 데이터셋으로 학습됐는데, 사람, 자동차, 의자, 컵 등 80개 종류의 물체를 인식할 수 있어요. 실행하면 각 물체에 바운딩 박스와 이름 라벨이 붙어서 나오는데, 처음 보면 "우와 이게 된다고?" 하고 놀라게 돼요.

📌 Step 3: 웹캠 실시간 인식으로 업그레이드

이미지 한 장 인식은 재미없죠? 웹캠을 연결해서 실시간 물체 인식을 해볼 거예요. OpenCV로 웹캠 스트림을 받아서 YOLOv8에 넘기면, 실시간으로 화면에 보이는 모든 물체를 인식해요.

여기서 핵심 팁! 모델 크기를 상황에 맞게 선택하세요. YOLOv8에는 nano, small, medium, large, xlarge 모델이 있어요. 로봇에 탑재할 거면 nano나 small 모델을 추천해요. 정확도는 약간 낮아지지만 속도가 훨씬 빨라요. Jetson 같은 엣지 디바이스에서는 nano 모델이 딱이에요.

📌 Step 4: 커스텀 물체 인식 — 내가 원하는 물체 학습시키기

기본 80개 클래스로 부족하면? 여러분만의 물체를 학습시킬 수 있어요! 예를 들어 로봇이 특정 부품을 분류해야 한다면, 해당 부품 이미지를 찍어서 라벨링하고 학습시키면 돼요.

라벨링에는 Roboflow나 CVAT 같은 무료 도구를 쓰면 편해요. 이미지 100~200장 정도만 라벨링하면 꽤 괜찮은 결과가 나오고, 앞서 소개한 Stable Diffusion으로 합성 데이터를 추가하면 더 좋아져요.

커스텀 학습도 한 줄 명령어로 가능해요. 학습 데이터 폴더 경로만 지정하면 알아서 학습이 시작돼요. GPU가 있으면 수십 분, 없어도 Colab에서 무료로 돌릴 수 있어요.

📌 Step 5: 로봇에 탑재하기

학습된 모델을 로봇에 올리는 방법은 간단해요. ONNX 포맷으로 내보내면 거의 모든 플랫폼에서 사용 가능해요. Jetson이면 TensorRT로 한 번 더 최적화하면 추론 속도가 2~3배 빨라져요.

로봇 카메라에서 인식된 물체 정보(종류, 위치, 크기)를 ROS2 토픽으로 퍼블리시하면, 다른 ROS2 노드에서 이 정보를 받아서 로봇 행동을 결정할 수 있어요. 예를 들어 "컵이 감지되면 집어"라는 로직을 추가하면 자동 물체 조작 로봇이 완성되는 거예요!

🎯 마무리 — 로봇의 눈을 뜨게 하세요

YOLOv8은 설치부터 실전 배포까지 로봇 비전의 진입 장벽을 확 낮춰준 도구예요. 예전엔 컴퓨터 비전 전문가만 할 수 있던 일을 이제 누구나 할 수 있거든요. 오늘 당장 시도해보세요. 여러분의 로봇에 눈이 생기는 순간, 가능성이 무한해져요! 다음 글에서는 인식된 물체를 로봇 팔로 집는 튜토리얼을 올릴게요. 👀