Cbam-Efficientnetv2-Enabled Fire Image Recognition Method in Detecting Tiny Targets

Bo Wang,Guozhong Huang,Haoxuan Li,Xiaolong Chen,Lei Zhang,XUEHONG GAO
DOI: https://doi.org/10.2139/ssrn.4188416
2022-01-01
SSRN Electronic Journal
Abstract:Image fire recognition is of great significance in fire prevention and loss reduction through early fire detection and warning. Aiming at the low accuracy of existing fire recognition and high error rate of tiny target detection, a fire recognition model based on an improved EfficientNetV2 is proposed in this paper. With the convolutional block attention module (CBAM), this study reconstructs an EfficientNetV2 by paying close attention to the target information from both channel and space dimensions. To improve the performance of classification labels, label smoothing (LS) is taken into account with a newly proposed cross-entropy loss function so that the generalization ability of the recognition model and high accuracy can be improved and achieved. To verify the effectiveness of the proposed method, extensive experiments are conducted to detect general and tiny targets. Furthermore, the proposed method is compared with the existing methods to illustrate its outperformance in detecting tiny targets. The results show that higher accuracy, precision, recall and F1-score were obtained in fire image recognition based on the proposed CBAM-EfficientNetV2 model in this study, with an accuracy up to 97.6%. Finally, the fire image results are visualized using the Grad-CAM technique, which makes the model more effective and more intuitive in detecting tiny targets.
What problem does this paper attempt to address?