记录一下用YOLOv3进行火焰识别 大学的毕业设计大致内容是火焰识别+手机APP+路径规划之类的东西。 其中火焰识别用到了YOLOv3模型,记录一下识别效果。 YOLO官方提供的数据集并没有火焰类别,因此只能自己训练一个。 大致用了550张照片,标了1000多个标注,经过6000次迭代,改了一下源码,让YOLO输出识别的火焰个数,最终效果如下: 启动YOLO 输出识别视频1 输出识别视频2 利用官方模型测试的行人计数1 2020-01-08 记录 #算法 #YOLO #计算机视觉