12 lines
588 B
YAML
12 lines
588 B
YAML
|
train:
|
||
|
|
- /workspace2/yolo_dataset/hat_head_2655_from_7581_yolov5x_auto_mark_person/train/images
|
||
|
|
- /workspace2/yolo_dataset/truck_2655_exp37best_auto_mark_person-hat-head/train/images
|
||
|
|
val:
|
||
|
|
- /workspace2/yolo_dataset/hat_head_2655_from_7581_yolov5x_auto_mark_person/valid/images
|
||
|
|
- /workspace2/yolo_dataset/truck_2655_exp37best_auto_mark_person-hat-head/valid/images
|
||
|
|
|
||
|
|
nc: 4
|
||
|
|
names: ['person','hat','head','truck']
|
||
|
|
|
||
|
|
# python train.py --data data/person_hat_head_truck.yaml --batch-size 32 --device=0 --epoch=100 --weights yolov5s.pt --imgsz 640 --cfg ./models/custom_yolov5s.yaml
|