魏云鹏, 吴开志, 俞子荣. 基于轻量化卷积神经网络的森林火灾烟雾检测算法[J]. 南昌航空大学学报(自然科学版), 2023, 37(2): 89-100. DOI: 10.3969/j.issn.2096-8566.2023.02.011
引用本文: 魏云鹏, 吴开志, 俞子荣. 基于轻量化卷积神经网络的森林火灾烟雾检测算法[J]. 南昌航空大学学报(自然科学版), 2023, 37(2): 89-100. DOI: 10.3969/j.issn.2096-8566.2023.02.011
Yun-peng WEI, Kai-zhi WU, Zi-rong YU. Forest Fire Smoke Detection Algorithm Based on Lightweight Convolutional Neural Network[J]. Journal of nanchang hangkong university(Natural science edition), 2023, 37(2): 89-100. DOI: 10.3969/j.issn.2096-8566.2023.02.011
Citation: Yun-peng WEI, Kai-zhi WU, Zi-rong YU. Forest Fire Smoke Detection Algorithm Based on Lightweight Convolutional Neural Network[J]. Journal of nanchang hangkong university(Natural science edition), 2023, 37(2): 89-100. DOI: 10.3969/j.issn.2096-8566.2023.02.011

基于轻量化卷积神经网络的森林火灾烟雾检测算法

Forest Fire Smoke Detection Algorithm Based on Lightweight Convolutional Neural Network

  • 摘要: 现有的基于卷积神经网络的森林火灾烟雾检测算法,存在烟雾特征提取结构过于复杂、烟雾多尺度特征融合方法过于繁琐、计算复杂度大以及应用场景单一等问题,而且其部署所需硬件配置高且难以适应多变的森林环境,这阻碍了其在森林防火领域的实际应用。为此,设计了一种基于轻量化卷积神经网络的森林火灾烟雾检测算法。首先,为提高烟雾特征提取的能力和速度,基于重参数化技术与跨阶段局部网络,提出一种轻量化烟雾特征提取结构。其次,基于简化的特征金字塔网络和路径聚合网络,设计出轻量化烟雾多尺度特征融合方法,实现不同尺度烟雾特征的高效融合。然后,提出一种烟雾检测后处理方法并增加类似烟雾图像进行算法模型训练,避免不同应用场景中非火灾烟雾图像和类似烟雾图像对检测算法的影响。最后,采用本文构建的烟雾图像数据集对算法进行验证。实验结果表明,本文算法相对于其它算法具有较高的检测精度和速度, F_1 分数达82.6%,AP值达54.5%,最高检测速度达869张/秒。

     

    Abstract: The forest fire smoke identification algorithm based on convolutional neural networks has some limitations, such as rather complicated smog feature extraction structure, tedious multi-scale feature fusion method, high computational complexity, limited application scenarios, high hardware requirements and difficult to adapt to the changing forest environment, which limit its application on the forest-fire prevention. To solve these problems, we proposed a lightweight convolutional neural network-based algorithm for detecting forest fire smoke. Firstly, based on re-parameterization technology and Cross Stage Partial Network, a lightweight structure for extracting smoke features was proposed to enhance the efficiency and velocity of smoke feature extraction. Secondly, a lightweight multi-scale smoke feature fusion method was developed utilizing the simplified Feature Pyramid Network and Path Aggregation Network to efficiently integrate smoke features across different scales. Then, a post-processing method for smoke detection was proposed, which involved the addition of similar smoke images to train the algorithm model, to mitigate the impact of non-fire smoke images and variations in application scenarios on detection accuracy. Finally, the algorithm was validated using the constructed smoke image dataset. The experimental results demonstrate that compared with other algorithms, the proposed algorithm is more accurate and faster with F1 up to 82.6%, AP value up to 54.5%, and the maximum detection speed up to 869 images per second.

     

/

返回文章
返回