记录一下用YOLOv3进行火焰识别

大学的毕业设计大致内容是火焰识别+手机APP+路径规划之类的东西。

其中火焰识别用到了YOLOv3模型，记录一下识别效果。

YOLO官方提供的数据集并没有火焰类别，因此只能自己训练一个。

大致用了550张照片，标了1000多个标注，经过6000次迭代，改了一下源码，让YOLO输出识别的火焰个数，最终效果如下：

启动YOLO

输出识别视频1

输出识别视频2

利用官方模型测试的行人计数1

利用官方模型测试的行人计数2