改进YOLOX-nano的火灾火焰烟雾检测

作者：汪子健; 高焕兵^*; 侯宇翔; 杜传胜; 贝太学

来源：计算机系统应用, 2023, 32(03): 265-274.

DOI：10.15888/j.cnki.csa.009000

• 火灾检测
• 注意力机制
• 软池化空间金字塔
• 损失函数
• YOLOX

针对在检测火焰和烟雾的火灾检测过程中存在火灾初期小目标难以检测的情况,本文提出了一种基于自然指数损失(eCIoU)的改进YOLOX-nano (ASe-YOLOX-nano)目标检测算法.首先,提出一种新的目标检测函数eIoU损失函数来替代传统IoU损失,解决在检测小目标时预测框和真实框易出现无交集的情况,及无法反应宽高影响等问题.其次,在网络模型中引入注意力模块,在网络初期模糊定位目标位置,提高网络后期对目标尤其是小目标检测的准确性.此外,本文还采用软池化空间金字塔池化结构提取不同尺寸的空间特征信息,可以提升模型对于空间布局和物体变性的鲁棒性,因此目标较小时也可以提取充足的特征,采用Mosaic增强技术预处理数据集,提升模型的泛化能力,以此进一步提高网络性能.通过目标数据集进行对比验证,其结果显示, mAP指标达到70.07%,比原模型提高了3.46%,火焰的准确率达到84.66%,烟雾的达到74.56%, FPS能够稳定在73,相对于传统YOLOXnano算法拥有更好的火灾检测能力.

• 出版日期2023
• 单位山东建筑大学; 机电工程学院