在上一篇[毕业设计之——yolov3搭建及测试](https://blog.csdn.net/fb1103/article/details/115376173)博客的基础上,进一步进行了查找数据集,训练,利用生成的权重文件,进行测试。
1、数据集的制作/搜索:
方法1:自己制作数据集——安装LabelImg老大难,目前安装任然还是问题(后期争取解决)
时间:2021年4月14日09:40:54,问题已解决:安装LabelImg成功
方法2:数据集的搜索:csdn,github,此处推荐一个不错的火灾数据集(作者已经标注好)
链接如下:火灾数据集链接
2、数据的训练:
电脑基本信息:GPU是:GTX1050
参考步骤:Bubbliiiing大佬的代码中README文件+自己百度
训练过程中截图:
问题:
<1>电脑温度过高:用了同学外置的风扇散热
<2>其中具体设置的参数还没搞懂!!!
3、训练完之后就是检测:
图片的检测结果:
视频的检测:火灾检测测试视频
后面主要的工作:优化程序,将烟雾和火灾结合,制作新的数据集,不同算法对比,论文