YOLOv7&YOLOv5&YOLOv3火焰和烟雾识别
1、本项目采用
YOLOv7
算法实现对烟雾和火焰的识别,在1000
多张火焰烟雾数据集中能训练得到,我们训练了两个YOLOv7
和YOLOv7x
两个模型,用于对比实验效果;此外还训练了yolov5s
和yolov3
两个模型用于对比实验,经评估我们得出了各个模型的评价指标,所有指标都是在同一个验证集上得到;本项目总体评估指标偏低,受数据集标注影响较大,后续可进行优化;
2、目标类别数据量:2
;类别名:fire
,smoke
3、项目代码版本:YOLOv5-6.0
;YOLOv3-3.9.6
4、本项目提供了数据集和上述几个训练好的模型的网盘链接可供下载
检测结果展示如下: