bert13069598 / yolov5

Exercise yolov5 using taekwon data in AIHub

yolov5

Exercise yolov5 using taekwon data sample in AIHub

Prepare data

taekwon data

Prerequisite

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

YOLOv5

yolov5

Preprocessing

I got ordered only use male samples.

So extract them from raw samples. Extract male

Reference

Train Custom Data

Adapt label files

Change json to txt

Adapt img files

Resize size to 640x640

Train & Validation & Test

Split label&img

Train

python train.py --img 640 --batch 16 --epochs 3 --data taekwon.yaml --weights yolov5s.pt

Run