复杂背景环境下基于SSDMobileNet深度学习模型的火焰图像识别研究

作者：卢鹏; 赵亚琴^*; 陈越; 孙一超; 徐媛

来源：火灾科学, 2020, 29(03): 142-149.

• 火焰识别
• 复杂环境
• 深度学习
• SSDMobileNet

针对现有的火灾火焰图像识别方法在光照和红花等类似火焰干扰的复杂环境下存在错检和漏检的问题,提出一种基于SSDMobileNet的复杂环境火焰区域标记方法。首先,将深度卷积神经网络SSD300的基础卷积网络VGG16替换为MobileNet网络,应用深度可分离卷积,降低网络参数,进而构建一种火焰图像检测的SSDMobileNet模型;然后,迁移第一次训练模型所有的卷积层参数,初始化新的待训练模型;最后,加入新的数据样本用于削弱光照、红花等干扰对象的影响。通过与SSD300、以及深度学习的目标检测算法Faster R-CNN和YOLOv3-tiny对比,实验结果表明,构建的火焰检测和火焰区域标记SSDMobileNet模型的综合性能优于Faster R-CNN和YOLOv3-tiny模型,更适用于实时火焰检测领域。

• 单位
  南京林业大学; 电子工程学院