Google Colaboratory 이용한 YOLOv5 데이터 학습

맥북 m1으로 돌리다가(CPU) 도저히 안 돌아가서 colab 이용함

걍 처음부터 코랩으로 돌리는 걸 추천한다,,,,!!

 

 

1. 구글 드라이브 -> Google Colaboratory 파일 생성

 

 

2. 내 드라이브 mount

from google.colab import drive
drive.mount('/content/drive')

- 둘 중 편한 방법으로 하면 됨

 

 

 

3. YOLOv5 설치

 

1) !git clone https://github.com/ultralytics/yolov5 

2) %cd yolov5

     !pip install -r requirements.txt

 

 

 

 

4. 그 후 과정은 똑같다. 파일들 및 실행 경로 설정만 내 드라이브로 다시 해주면 됨

https://sssbin.tistory.com/163 

학습 완.

**** 런타임 -> 런타임 유형 변경 에서 꼭 GPU로 변경해주기!!!

 

 

5. 검증하기

!python val.py --data "/content/drive/MyDrive/GP/coca/data.yaml" --img 1280 --weights "/content/yolov5/runs/train/coca2/weights/best.pt"

 

 

 

6. 예측하기

!python detect.py --img 1280 --weights "/content/yolov5/runs/train/coca2/weights/best.pt" --source "/content/drive/MyDrive/GP/coca/testCoca.jpg"

와앙

 

 

***

!zip -r /content/yolov5.zip /content/yolov5

-> 집파일로 만들어서 내 PC에 다운 가능