라즈베리파이[Raspberry Pi] YOLOv5 실시간 객체 인식 및 PiCamera 세팅

카메라 연결

• 당연히 카메라가 있어야 합니다.
• 저는 Picamera V2를 사용했으며 같은 모델이라면 아래 사진을 참고하여 연결해주시면 됩니다.

 

 

라즈베리파이 카메라 세팅

• 아래 명령어를 입력합니다.

sudo raspi-config

 

• Interface options 에 들어갑니다

 

• Legacy Camera에서 엔터를 입력합니다.

 

• <예>를 선택한 후 재부팅 합니다.

 

 

YOLOv5 실시간 객체 인식

• 아래 링크를 참고하여 YOLOv5 모델을 설치하고 세팅을 먼저 해주어야 합니다.

https://chlduswns99.tistory.com/48

라즈베리파이[Raspberry Pi] YOLOv5 기본 세팅

 

 

• 다음 아래 코드를 이용하여 실시간 객체 인식 테스트를 해주면 됩니다. 

 python3 detect.py --source 0 --weights yolov5s.pt --conf 0.25

 

• --source 0 : 연결할 카메라를 뜻합니다. 카메라가 2개 연결되어 있다면 0과 1로 구분되고, 하나만 연결되어 있다면 0을 입력합니다.
• --weights yolov5s.pt : 모델을 입력합니다. 저는 기본 모델인 yolov5s 모델을 사용하였고 만약 커스텀 모델이 있다면 해당 경로를 입력해주면 됩니다.
• --conf 0.25: 해당 객체라고 인식할 확률을 입력합니다. 25% 이상이라면 해당 객체라고 예측하고 출력합니다.

 

 

• yolov5s 모델이 현재 laptop을 80% 확률로 예측하고 있는걸 볼 수 있습니다.