节点文献
基于轻量化YOLOv4的火灾检测识别算法
Fire detection and recognition algorithm based on lightweight YOLOv4
【摘要】 为了满足火灾实时蔓延对检测速度和准确率的更高要求,在YOLOv4的基础上,提出了一种轻量化火灾检测方法。将改进的MobileNetV3作为主干特征提取网络来降低模型复杂度,提高火灾检测速度,并引入高效通道注意力(ECA)机制模块,有效地捕获了跨通道交互,增强了对火灾目标区域的重点关注。在加强特征提取网络部分,采用了加权双向特征金字塔(BiFPN)网络结构,不增加额外参数的同时,融合了更多不同尺度的特征,对不同大小火灾区域的检测精度有了显著提升。实验结果表明:所提方法具有较好的火灾检测效果,在自建的数据集上,平均精度达到了86.4%,检测速度达到了58 fps,相较于原模型,分别提升了2.3%和24 fps,同时模型大小缩减了79%。
【Abstract】 In order to meet the higher requirements of detection speed and accuracy of real-time spread of fire, a lightweight fire detection method based on YOLOv4 is proposed.The improved MobileNetV3 is used as the backbone feature extraction network to reduce the complexity of the model and increase the speed of fire detection, and introduce the efficient channel attention(ECA)mechanism module, which effectively captures cross-channel interaction and enhances the focus on the fire target area.In the enhanced feature extraction network part, the BiFPN weighted two-way feature pyramid structure is adopted, which fuses more features of different scales without adding additional parameters, and the detection precision of fire areas of different sizes has been significantly improved.The experimental results show that the proposed method has a good fire detection effect.On the self-built dataset, the average precision reaches 86.4 %,and the detection speed reaches 58 fps.Compared with the original model, it is improved by 2.3 % and 24 fps, and at the same time, the size of the model is reduced by 79 %.
【Key words】 fire detection; deep learning; attention mechanism; multi-scale feature fusion; depthwise separable convolution;
- 【文献出处】 传感器与微系统 ,Transducer and Microsystem Technologies , 编辑部邮箱 ,2023年08期
- 【分类号】TP183;TP391.41;X932
- 【下载频次】85