빠른 시작 가이드: 라즈베리 파이 및 Pi 카메라( YOLOv5 및 YOLOv8

이 종합 가이드는 Pi 카메라를 사용해 라즈베리파이에서 물체 감지 모델( YOLO )을 빠르게 구축할 수 있도록 도와드립니다. 이 가이드는 학생, 취미 활동가, 전문가 등 누구나 30분 이내에 시작하고 실행할 수 있도록 설계되었습니다. 이 가이드의 지침은 설정 문제를 최소화하기 위해 엄격한 테스트를 거쳤으므로 특정 프로젝트에 YOLO 을 활용하는 데 집중할 수 있습니다.

Watch: 라즈베리 파이 5 업데이트 및 개선 사항.

전제 조건

• 라즈베리 파이 3, 4, 5
• Pi 카메라
• 64비트 라즈베리 파이 운영 체제

CSI 케이블을 통해 Pi 카메라를 라즈베리파이에 연결하고 64비트 라즈베리파이 운영체제를 설치합니다. 다음 명령으로 카메라를 확인합니다:

libcamera-hello

카메라에서 비디오 피드가 표시되어야 합니다.

YOLO 버전 선택: YOLOv5 또는 YOLOv8

이 가이드는 다음 중 하나로 시작할 수 있는 유연성을 제공합니다. YOLOv5 또는 YOLOv8. 두 버전 모두 고유한 장점과 사용 사례가 있습니다. 선택은 여러분의 몫이지만, 이 가이드의 목표는 빠른 설정뿐 아니라 향후 객체 감지 작업을 위한 견고한 토대를 마련하는 것임을 기억하세요.

하드웨어 사양: 한눈에 보기

정보에 입각한 하드웨어 결정을 돕기 위해 아래 표에 라즈베리파이 3, 4, 5의 주요 하드웨어 사양을 요약해 놓았습니다:

원활한 설정 과정을 위해 사용 중인 라즈베리파이 모델에 맞는 지침을 따르시기 바랍니다.

빠른 시작 YOLOv5

이 섹션에서는 라즈베리파이에서 Pi 카메라로 YOLOv5 를 설정하는 방법을 간략하게 설명합니다. 이 단계는 라즈베리파이 OS 불스아이에 도입된 libcamera 카메라 스택과 호환되도록 설계되었습니다.

필요한 패키지 설치

1. 라즈베리 파이를 업데이트합니다:

   sudo apt-get update
   sudo apt-get upgrade -y
   sudo apt-get autoremove -y
   

2. YOLOv5 리포지토리를 복제합니다:

   cd ~
   git clone https://github.com/Ultralytics/yolov5.git
   

3. 필요한 종속성을 설치합니다:

   cd ~/yolov5
   pip3 install -r requirements.txt
   

4. 라즈베리 파이 3의 경우 호환되는 버전의 PyTorch 및 Torchvision을 설치합니다(라즈베리 파이 4의 경우 건너뛰기):

   pip3 uninstall torch torchvision
   pip3 install torch==1.11.0 torchvision==0.12.0
   

수정 detect.py

SSH 또는 CLI 를 통해 TCP 스트림을 사용하려면 다음에서 약간의 수정이 필요합니다. detect.py.

1. 열기 detect.py:

   sudo nano ~/yolov5/detect.py
   

2. 찾기 및 수정 is_url 라인을 사용하여 TCP 스트림을 수락합니다:

   is_url = source.lower().startswith(('rtsp://', 'rtmp://', 'http://', 'https://', 'tcp://'))
   

3. 댓글로 view_img 라인:

   # view_img = check_imshow(warn=True)
   

4. 저장하고 종료합니다:

   CTRL + O -> ENTER -> CTRL + X
   

Libcamera로 TCP 스트림 시작하기

1. TCP 스트림을 시작합니다:

   libcamera-vid -n -t 0 --width 1280 --height 960 --framerate 1 --inline --listen -o tcp://127.0.0.1:8888
   

다음 단계를 위해 이 터미널 세션을 계속 실행하세요.

YOLOv5 추론 수행

1. YOLOv5 탐지를 실행합니다:

   cd ~/yolov5
   python3 detect.py --source=tcp://127.0.0.1:8888
   

빠른 시작 YOLOv8

대신 YOLOv8 설정에 관심이 있는 경우 이 섹션을 따르세요. 단계는 매우 유사하지만 YOLOv8 의 특정 요구 사항에 맞게 조정되었습니다.

필요한 패키지 설치

1. 라즈베리 파이를 업데이트합니다:

   sudo apt-get update
   sudo apt-get upgrade -y
   sudo apt-get autoremove -y
   

2. 설치 ultralytics Python 패키지입니다:

   pip3 install ultralytics
   

3. 재부팅합니다:

   sudo reboot
   

Libcamera로 TCP 스트림 시작하기

1. TCP 스트림을 시작합니다:

   libcamera-vid -n -t 0 --width 1280 --height 960 --framerate 1 --inline --listen -o tcp://127.0.0.1:8888
   

YOLOv8 추론 수행

YOLOv8 를 사용하여 추론을 수행하려면 다음 Python 코드 스니펫을 사용할 수 있습니다:

from ultralytics import YOLO

model = YOLO('yolov8n.pt')
results = model('tcp://127.0.0.1:8888', stream=True)

while True:
    for result in results:
        boxes = result.boxes
        probs = result.probs

다음 단계

라즈베리파이에 YOLO 를 성공적으로 설치하신 것을 축하드립니다! 더 많은 학습과 지원을 원하시면 Ultralytics 와 카슈미르 월드 재단을 방문하세요.

감사 및 인용

이 가이드는 멸종 위기 종 보존을 위해 YOLO 을 사용하는 단체인 카슈미르 월드 재단을 위해 다안 엘팅크가 처음에 만들었습니다. 물체 감지 기술 분야에서 선구적인 작업과 교육에 중점을 둔 이 단체의 공로를 인정합니다.

카슈미르 월드 재단의 활동에 대한 자세한 내용은 웹사이트에서 확인할 수 있습니다.

댓글