Yolov5 Retrain

git clone https://github.com/ultralytics/yolov5
cd yolov5
pip install -U -r requirements.txt # -q
python3 -c "from yolov5.utils.google_utils import gdrive_download; gdrive_download('1n_oKgR81BJtqk75b00eAjdv03qVCQn2f','coco128.zip')" 

# yaml file
path: ../datasets/person  # dataset root di
train: images/train2017  # train images (relative to 'path') 128 images
val: images/train2017  # val images (relative to 'path') 128 images
nc: 1  # number of classes
names: [ 'person' ]

| ---- yolov5
| ---- datasets
|       | --- images
|                |  --- train
|                         | --- train_1.jpg
|                         | --- ...
|                         | --- train_n.jpg 
|                |  --- val
|                         | --- val_1.jpg 
|                         | --- ...
|                         | --- val_n.jpg 
|       | --- labels 
|                |  --- train
|                         | --- train_1.txt
|                         | --- ...
|                         | --- train_n.txt
|                |  --- val
|                         | --- val_1.txt
|                         | --- ...
|                         | --- val_n.txt

pip install labelimg
labelimg

python train.py --img 640 --batch 16 --epochs 100 --data ./data/person.yaml --cfg ./models/yolov5s.yaml

python detect.py --weight person.pt --source 0