基于yolov5的烟火检测
前面一系列安装环境配置问题不在此详细说明 以下只讨论目前阶段所遇到的问题已经解决方案
预先权重使用的是5s
数据集是使用的讯飞平台下载的烟火图片 已经做好了标注 在跑完50轮 和100轮训练后 mAP比较低并不理想
现在考虑一下问题
1.数据集不够高清或者是图像自身所带的标注不是特别好
2.训练轮次数不够
有群友说我的已经出现了过拟合现象 所以大概率推断数据集有一定问题
以下是用我自己的权重以及下载的最好权重做对比
基于yolov5的烟火检测
最新推荐文章于 2023-06-11 01:46:11 发布