• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2023, Vol. 45 ›› Issue (10): 1884-1890.

• Artificial Intelligence and Data Mining • Previous Articles     Next Articles

A forest fire image early warning detection method based on probabilistic two-stage CenterNet2

LI Bao-min,WANG Xiao-peng,SUN Qian-rong,ZHANG Jun-ping   

  1. (School of Electronic and Information Engineering,Lanzhou Jiaotong University,Lanzhou 730070,China)
  • Received:2022-10-21 Revised:2023-03-27 Accepted:2023-10-25 Online:2023-10-25 Published:2023-10-17

Abstract: Timely warning of forest fire plays a crucial role in forest protection. Due to the complex background of forest fireworks and many interference factors, the detection accuracy and efficiency are affected. Therefore, a forest fire image detection method based on CenterNet2 is proposed. The lightweight backbone network VoVNetV2 combined with asymmetric convolution kernel is used to improve the feature extraction ability and detection speed. Meanwhile, an attention mechanism eSE (Effective Squeeze and Extraction) is introduced into the weighted bidirectional feature pyramid network for feature fusion, so as to improve the accuracy of small target detection. Then, SIoU loss function is used to improve the effect of target box regression. The simulation results show that the method can accurately detect forest fire in real time, and the false rate is low. 


Key words: forest fires, asymmetric convolution kernel, attention mechanism, SIoU loss function