YoloV6:YoloV6+Win10---训练自己得数据集

发布时间:2022-08-29 07:30

yolov6刚出来得那会我也跑了下,问题实在太多了,各种得bug,就暂时停了会,后面有时间了在看看就重新git下跑了跑。据说YoloV7也出来是AB大神团队出品得,不是某个人所谓得YoloV7....

GitHub - meituan/YOLOv6: YOLOv6: a single-stage object detection framework dedicated to industrial applications.YOLOv6: a single-stage object detection framework dedicated to industrial applications. - GitHub - meituan/YOLOv6: YOLOv6: a single-stage object detection framework dedicated to industrial applications.https://github.com/meituan/YOLOv6

1、数据集

我用的是自己得数据集,先看下数据集格式吧。

val和train存放得是图片,labels存放的是train  val文件夹,里面放的是txt

YoloV6:YoloV6+Win10---训练自己得数据集_第1张图片

YoloV6:YoloV6+Win10---训练自己得数据集_第2张图片

YoloV6:YoloV6+Win10---训练自己得数据集_第3张图片

 同时在查看下配置文件,我的数据集 只有两个类别

YoloV6:YoloV6+Win10---训练自己得数据集_第4张图片

 此处还要改下dataset.py里面得一处;YoloV6:YoloV6+Win10---训练自己得数据集_第5张图片

 

2、训练

训练命令:

python tools/train.py --batch 6 --conf configs/yolov6s_finetune.py --data data/MyDataCoCo.yaml --device 0  --epochs 800 --workers 0

我是在自己本机训练的,所以bs设置得小点

训练结束,大家看到,map不为0了,那是我们使用得是yolov6s_finetune.py  如果大家使用的是yolov6s.py,相当于没有加载与训练模型,具体大家可以去Yolov6的github  issues去看看。

https://github.com/meituan/YOLOv6/issues/205#issuecomment-1173409870

YoloV6:YoloV6+Win10---训练自己得数据集_第6张图片

 800轮训练时间将近19个小时。。。。。。。。。

3、测试

测试代码:

python tools/infer.py --weights runs/train/exp/weights/best_ckpt.pt --source CiWaData/val --yaml data/MyDataCoCo.yaml

测试结果:

YoloV6:YoloV6+Win10---训练自己得数据集_第7张图片 YoloV6:YoloV6+Win10---训练自己得数据集_第8张图片

 

4、ONNX导出

导出命令:

export_onnx.py --weights runs/train/exp/weights/best_ckpt.pt --img 640 --batch 1

 

 

 

ItVuer - 免责声明 - 关于我们 - 联系我们

本网站信息来源于互联网,如有侵权请联系:561261067@qq.com

桂ICP备16001015号