基于YOLOv3-tiny的火焰目标检测算法

徐岩; 李永泉; 郭晓燕; 韩立苏; 刘巧玲

doi:10.16452/j.cnki.sdkjzk.2022.06.010

山东科技大学学报（自然科学版） ›› 2022, Vol. 41 ›› Issue (06) : 95 -103. DOI: 10.16452/j.cnki.sdkjzk.2022.06.010

基于YOLOv3-tiny的火焰目标检测算法

徐岩, 李永泉, 郭晓燕, 韩立苏, 刘巧玲

作者信息 +

Author information +

文章历史 +

PDF (2840K)

摘要

基于YOLOv3-tiny网络提出一种满足实时性需求的火焰目标检测算法，解决现有一般算法在自制火焰数据集上检测效果不佳的问题。通过构建CSP-MobileNetV2结构并引入到骨干网络池化层替换原始卷积层，提升网络的特征提取能力；引入选核注意力网络(SKNet),使卷积核对不同尺寸的火焰目标实现自适应调整，增强了对火焰特征的提取能力；引入金字塔池化模块(PPM),将具有不同感受野的特征图进行融合，实现多级特征信息融合互补，提高特征表达能力。实验结果表明，改进算法在自制火焰数据集上的平均检测精度均值达到84.3%,比原网络提高4.1%。