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

最新推荐文章于 2025-04-07 20:57:15 发布

Mrs.Gril

最新推荐文章于 2025-04-07 20:57:15 发布

阅读量1.3k

点赞数 1

分类专栏： AI-训练+部署文章标签： YoloV6 深度学习人工智能计算机视觉

本文链接：https://blog.csdn.net/zhangdaoliang1/article/details/125662795

版权

AI-训练+部署专栏收录该内容

57 篇文章

订阅专栏

本文介绍了作者使用自己的数据集对YOLOv6进行训练的过程，包括数据集格式、训练配置、训练过程以及训练后的测试和ONNX模型导出。在训练过程中，作者提到Yolov6s_finetune.py使得模型在训练后得到非零的mAP，而使用yolov6s.py则会导致mAP为零。此外，文章还分享了训练时间和测试结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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

此处还要改下dataset.py里面得一处；

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

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

3、测试

测试代码:

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

测试结果:

4、ONNX导出

导出命令：

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