A high-precision lightweight smoke detection model based on SE attention mechanism

Fajun Yu,Wangjin Xia,Ruikai Hong,Huanqin Wang
DOI: https://doi.org/10.1109/ICCECE54139.2022.9712739
2022-01-14
Abstract:a high-precision and fast outdoor smoke detection method is proposed. The existing smoke detection model can not meet the real-time requirements on the equipment with low computing power, so a Yolo-attention lightweight model is adopted in this paper. The backbone adopts shufflenetv2, which can reduce the amount of parameters and calculation of the model. In order to achieve high-precision detection of outdoor smoke, we collected and established a data set of outdoor smoke combustion, and used data enhancement technology and attention mechanism. Mosaic data is used to enhance the effect of small target detection, and attention mechanism is used to enhance the information sharing between model channels. The experimental results show that the detection accuracy can be improved by 9. 8% after data augmentation. Coupled with the attention mechanism, the detection accuracy can be increased by 1.3%. At the same time, the amount of calculation of the improved model is 1. 1 G FLOPs and the size is only 4.8M, which has been greatly improved. This paper proposes a high-precision lightweight model for outdoor smoke detection, which provides a possible method for outdoor smoke detection such as forest fire prevention and motor vehicle smoke distinguishing.
Computer Science,Environmental Science
What problem does this paper attempt to address?