train: /workspace2/yolo_dataset/hel/train/images 
val: /workspace2/yolo_dataset/hel/valid/images 

nc: 3
names: ['person','hat','head']

# python train.py --data data/person_hat_head.yaml --batch-size 32 --device=0 --epoch=150 --weights yolov5s.pt --imgsz 640  --cfg ./models/custom_yolov5s_3.yaml