yolov5_v6_plate_heading

• 集成yolov5(v6.0), yolox, 注意力机制， 和repvgg结构

• 添加了多头检测代码，使用train_plate.py文件进行训练

• 添加了检测+关键点代码，使用train_key.py文件进行训练

# 多头训练
$ python train_plate.py --data data/mydata.yaml --batch 256 --epochs 200 --weights weights/yolov5s.pt   --imgsz 416  --device '0,1'  --cfg models/yolov5s_plate.yaml  --hyp data/hyps/palte_head.yaml 

# 关键点训练
$ python train_key.py --data data/yolov5s_key.yaml --batch 256 --epochs 200 --weights weights/yolov5s.pt   --imgsz 416  --device '0'  --cfg models/yolov5s_key.yaml  --hyp data/hyps/hyp.key.yaml 

If you use multi-gpu. It's faster several times:

$ python -m torch.distributed.launch --nproc_per_node 2 train_plate.py --sync-bn