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

 

YOLOv5 디렉토리 클론

• 먼저 YOLOv5 깃허브에서 모델을 가져와야 합니다.
• 라즈베리파이 터미널 창에서 아래 코드를 이용하여 파일을 클론합니다.

git clone https://github.com/ultralytics/yolov5

(저는 따로 학습 시켜놓은 모델이 있어 WinSCP를 이용하여 파일을 옮겼습니다.)

 

 

파일을 잘 불러왔는지 확인

 

• 아래 코드를 입력하여 파일이 잘 불러졌는지 확인합니다.
• 미리 설치하지 않았다면 torch 모듈이 없다고 나오는게 정상입니다.

## cd 명령어를 이용해 먼저 yolo 디렉토리로 이동해주어야 합니다.
python3 detect.py --source data/images --weights yolov5s.pt --conf 0.25

 

종속성 설치

• torch 모듈을 사용하기 위한 종속성 설치입니다. 터미널에서 아래 코드를 사용해 설치합니다.

sudo apt-get install python3-pip libjpeg-dev libopenblas-dev libopenmpi-dev libomp-dev

 

 

PyTorch & 필요 라이브러리 설치

• PyTorch의 경우 버전에 유의해야 합니다.
• 본인은 Pytorch 호환 오류로 인해 라즈베리파이 OS를 bullseye 버전으로 설치했습니다. (원래는 BookWorm을 사용하였음)
• 만약 OS 버전이 다르다면 호환되는 버전을 찾아 설치해야 합니다.
• 아래 링크에서 맞는 버전 확인이 가능합니다. (참고로 라즈베리파이 OS의 가장 최신 버전인 BookWorm에 대한 정보는 아직 없습니다.)

https://github.com/Qengineering/PyTorch-Raspberry-Pi-64-OS

GitHub - Qengineering/PyTorch-Raspberry-Pi-64-OS: PyTorch installation wheels for Raspberry Pi 64 OS

 

 

• 본인은 torch 1.9.0 버전과 torchvision 0.10.0 버전을 사용했습니다.

 

맞는 버전을 결정하였다면 위 깃허브 주소에서 구글 드라이브로 다운 받아 설치를 진행해줍니다.

 

• 먼저 구글 드라이브 이용을 위해 gdown 라이브러리를 이용합니다.

sudo pip3 install gdown

 

• 다음 아래 코드를 순차적으로 입력해 설치합니다.

gdown https://drive.google.com/uc?id=1p9Sp5YmCJwIkmQIWkpuIkE4NgeYPOr-U

sudo -H pip3 install torch-1.9.0a0+gitd69c22d-cp39-cp39-linux_aarch64.whl

gdown https://drive.google.com/uc?id=13fKq1V-zLbLBWPw3zP4EZxyMXyOiJWPi

sudo -H pip3 install torchvision-0.10.0a0+300a8a4-cp39-cp39-linux_aarch64.whl

(만약 링크가 존재하지 않는다면 직접 구글 드라이브에 파일을 넣어서 다운 받으세요)

 

• 다음 필요 라이브러리를 설치합니다.

pip install -r requirements.txt

 

 

간단한 테스트

 

• 아래 코드를 실행시켜 결과가 저장되었는지 확인합니다.
• 실행이 완료되었다면 맨 아래 줄에 결과물의 저장 위치를 알려준다. 해당 디렉토리로 이동하여 결과물을 확인하면 됩니다.

python3 detect.py --source data/images/ --weights yolov5s.pt --conf 0.25