8 lines
320 B
YAML
8 lines
320 B
YAML
|
train: /workspace2/yolo_dataset/2025-02-07-hat-head/train/images
|
||
|
|
val: /workspace2/yolo_dataset/2025-02-07-hat-head/valid/images
|
||
|
|
|
||
|
|
nc: 3
|
||
|
|
names: ['person', 'hat', 'head']
|
||
|
|
|
||
|
|
# python train.py --data data/hat_head.yaml --batch-size 32 --device=0 --epoch=300 --weights yolov5s.pt --imgsz 640 --cfg ./models/custom_yolov5.yaml
|