# Forest fire detection and recognition method based on improved YOLOv5-ACE algorithm

## TL;DR

This paper introduces an improved YOLOv5-ACE algorithm for faster and more accurate detection of forest fires.

## Contribution

The novel contribution is integrating CBAM, ASPP, ShuffleNet v2, and ViT to enhance small target detection and model adaptability.

## Key findings

- The detection accuracy increased by 11.5%, reaching 92.3%.
- Recall improved by 6.8%, reaching 91.6%.

## Abstract

Currently, forest fires have become a major fire safety issue. To detect forest fires and optimize the accuracy, a forest fire detection and recognition model based on an improved YOLOv5-ACE algorithm is proposed. In response to the difficulties of small target detection in forest fires, poor adaptability to complex backgrounds, and deployment limitations of edge devices, the CBAM and the ASPP multi-scale feature extraction module are introduced to enhance the ability of target feature capture and small target detection. The algorithm is lightweight by combining the grouped convolution of ShuffleNet v2 and the global dependency capture of ViT, while improving the positioning accuracy and anti-interference ability. Compared with the traditional YOLOv5, the detection accuracy has increased by 11.5%, ultimately reaching 92.3%, and the recall has increased by 6.8% to 91.6%. Through hypothesis testing, all performance improvements have statistical significance (p < 0.05). The proposed method can detect forest fires more quickly and accurately, which has good guiding significance for preventing the occurrence of forest fires.