
基于YoloV5-V6系列,Train-Palate增加多头检测功能,Train-Key加入关键点检测算法
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:Train-Palate在YoloV5-V6基础上新增了多目标检测能力,而Train-Key则引入了先进的关键点检测算法,两者共同推动了实时物体识别与定位技术的革新。
使用说明在zip压缩包的README文件中,请仔细阅读。本项目集成了yolov5(v6.0)、yolox,并添加了注意力机制和repvgg结构,支持多头检测功能。具体训练方法如下:
- 多头检测训练:请运行`train_plate.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
```
- 关键点检测训练:请运行`train_key.py`文件进行训练。
- 命令示例:
```
$ python train_key.py --data data/yolov5s_key.yaml --batch 256 --epochs 200 --weights weights/yolov5s.pt --imgsz 416 --device 0
```
全部评论 (0)
还没有任何评论哟~


