计算机与现代化 ›› 2021, Vol. 0 ›› Issue (11): 61-66.

• 图像处理 • 上一篇    下一篇

融合注意力机制的无锚点森林火灾检测算法

  

  1. (1.新疆师范大学计算机科学技术学院,新疆乌鲁木齐830054;
    2.大连理工大学计算机科学与技术学院,辽宁大连116024)

  • 出版日期:2021-12-13 发布日期:2021-12-13
  • 作者简介:陆雅诺(1996—),男,安徽铜陵人,硕士研究生,研究方向:计算机视觉,E-mail: xjnu@foxmail.com; 通信作者:陈炳才(1978—),男,福建漳平人,教授,博士,研究方向:图像处理,物联网,无线通信与网络,E-mail: china@dlut.edu.cn。
  • 基金资助:
    国家自然科学基金资助项目(61961040, 61771087); 新疆维吾尔自治区“天山青年计划”(2018Q024); 自治区区域协同创新专项(科技援疆计划)(2020E0247, 2019E0214)

An Anchor-free Forest Fire Detection Algorithm Incorporating Attention Mechanism

  1. (1. College of Computer Science and Technology, Xinjiang Normal University, Urumqi  830054, China;
    2. School of Computer Science and Technology, Dalian University of Technology, Dalian   116024, China)
  • Online:2021-12-13 Published:2021-12-13

摘要: 森林火灾、野火是一个重大的自然灾害问题,每年全球各地植被都会受到严重的破坏。为了提高森林火灾的防控精度,针对传统方法具有火灾背景复杂、准确率低、效率低等问题,本文提出一种基于CenterNet的森林火灾检测算法。CenterNet作为一种无锚的方法,将目标定义为一个点,通过关键点估计定位目标的中心点,可以有效避免小目标的漏检。同时基于高效深层特征提取网络ResNet50,融合ECA模块以抑制无用信息,增加模型的特征提取能力。在公开森林火灾数据集上进行实验表明,与其他算法相比,本文提出的森林火灾检测算法误检率低,识别精度达到92.39%,F1值为0.86,Recall值为79.75%,FPS为43.31。本文提出的方法检测精度高,可满足实时检测森林火灾和实施精准施救的要求。

关键词: 注意力机制, 无锚点检测, 森林火灾

Abstract: Forest fire and wildfire are major natural disaster problems, and vegetation is severely damaged all over the world every year. In order to improve the accuracy of forest fire prevention and control, aiming at the problems of complex fire background, low accuracy and low efficiency of traditional methods, this paper proposes a forest fire detection algorithm based on CenterNet. As an anchorless method, CenterNet defines a target as a point and locates the centroid of the target by key point estimation, which can effectively avoid the missed detection of small targets. At the same time, based on an efficient deep feature extraction network, ResNet50, it incorporates an ECA module to suppress useless information and increase the feature extraction capability of the model.Experiments conducted on public forest fire datasets show that compared with other arithmetic methods, the forest fire detection algorithm proposed in this paper has a low false detection rate and a recognition accuracy of 92.39% with a F1 value of 0.86, a Recall value of 79.75%, and a FPS of 43.31.The proposed method has a high detection accuracy and achieves real-time detection of forest fires and implementation of accurate rescue.

Key words: attention mechanism, anchorless detection, forest fire