yolov5 커스텀하기!

google colab에서 돌렸습니다!

 

학습 데이터 준비

- roboflow 등 준비된 데이터 이용

- 직접 라벨링 ,,

 

yolov5 다운로드

%cd /content
!git clone https://github.com/ultralytics/yolov5.git

- yolov5 requirements 다운로드 (yolov5 학습에 필요한 라이브러리 다운로드)

%cd /content/yolov5/
! pip install -r requirements.txt

사전준비

%cd /
from glob import glob

train_img_list = glob('/content/[roboflow 프로젝트명]/train/images/*.jpg')
val_img_list = glob('/content/[roboflow 프로젝트명]/valid/images/*.jpg')

with open('content/[roboflow 프로젝트명]/train.txt', 'w') as f:
  f.write('\n'.join(train_img_list)+'\n')

with open('content/[roboflow 프로젝트명]/val.txt', 'w') as f:
  f.write('\n'.join(val_img_list)+'\n')

import yaml

with open('/content/[roboflow 프로젝트명]/data.yaml', 'r') as f:
  data = yaml.load(f, Loader=yaml.FullLoader)

data['train'] = '/content/[roboflow 프로젝트명]/train.txt'
data['val'] = '/content/[roboflow 프로젝트명]/val.txt'

with open('/content/[roboflow 프로젝트명]/data.yaml', 'w') as f:
  yaml.dump(data, f)

print(data)

학습 후 로컬에 다운로드

%cd /content/yolov5/

!python train.py --img 416 --batch 16 --epochs 150 --data /content/[roboflow 프로젝트명]/data.yaml --cfg ./models/yolov5s.yaml --weights yolov5s.pt --name [저장할이름]

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

from google.colab import files
files.download('/content/yolov5.zip')

- 커스텀 모델 로컬(장고)에서 사용해보기

https://security-guide.tistory.com/79

Use Customized AI Model(yolov5) in Django