yolov5 coco data set training

1. 먼저 yolov5를 다운한다.

yolov5관련된 lib도 설치한다.

!git clone https://github.com/ultralytics/yolov5 # clone repo
%cd yolov5
!pip install -U -r requirements.txt

2. yolov5/data/coco.yaml수정

path: ./datasets/coco

경로 수정

 

3. train 실행

!python train.py --data coco.yaml --epoch 1

 

 

 

참고 사이트 :

https://stackoverflow.com/questions/55556965/importing-coco-datasets-to-google-colaboratory

Importing COCO datasets to google colaboratory