Abstract:Forest fires occur frequently around the world, causing serious economic losses and human casualties. Deep learning techniques based on convolutional neural networks (CNN) are widely used in the intelligent detection of forest fires. However, CNN-based forest fire target detection models lack global modeling capabilities and cannot fully extract global and contextual information about forest fire targets. CNNs also pay insufficient attention to forest fires and are vulnerable to the interference of invalid features similar to forest fires, resulting in low accuracy of fire detection. In addition, CNN-based forest fire target detection models require a large number of labeled datasets. Manual annotation is often used to annotate the huge amount of forest fire datasets; however, this takes a lot of time. To address these problems, this paper proposes a forest fire detection model, TCA-YOLO, with YOLOv5 as the basic framework. Firstly, we combine the Transformer encoder with its powerful global modeling capability and self-attention mechanism with CNN as a feature extraction network to enhance the extraction of global information on forest fire targets. Secondly, in order to enhance the model's focus on forest fire targets, we integrate the Coordinate Attention (CA) mechanism. CA not only acquires inter-channel information but also considers direction-related location information, which helps the model to better locate and identify forest fire targets. Integrated adaptively spatial feature fusion (ASFF) technology allows the model to automatically filter out useless information from other layers and efficiently fuse features to suppress the interference of complex backgrounds in the forest area for detection. Finally, semi-supervised learning is used to save a large amount of manual labeling effort. The experimental results show that the average accuracy of TCA-YOLO improves by 5.3 compared with the unimproved YOLOv5. TCA-YOLO also outperformed in detecting forest fire targets in different scenarios. The ability of TCA-YOLO to extract global information on forest fire targets was much improved. Additionally, it could locate forest fire targets more accurately. TCA-YOLO misses fewer forest fire targets and is less likely to be interfered with by forest fire-like targets. TCA-YOLO is also more focused on forest fire targets and better at small-target forest fire detection. FPS reaches 53.7, which means that the detection speed meets the requirements of real-time forest fire detection.

CF-YOLO: a capable forest fire identification algorithm founded on YOLOv7 improvement

Multi-Scale Forest Fire Recognition Model Based on Improved YOLOv5s

A Small Target Forest Fire Detection Model Based on YOLOv5 Improvement

Forest fire detection algorithm based on Improved YOLOv5

A Semi-Supervised Method for Real-time Forest Fire Detection Algorithm Based on Adaptively Spatial Feature Fusion

Research and Implementation of Forest Fire Detection Algorithm Improvement

Fire-PPYOLOE: An Efficient Forest Fire Detector for Real-Time Wild Forest Fire Monitoring

YOLOv7-FIRE: A tiny-fire identification and detection method applied on UAV

Multi Scale Forest Fire Small Target Detection Technology

Fire-YOLO: A Small Target Object Detection Method for Fire Inspection

YOLOFM: an improved fire and smoke object detection algorithm based on YOLOv5n

An effective graph embedded YOLOv5 model for forest fire detection

A Deep Learning Method based on SRN-YOLO for Forest Fire Detection

FL-YOLOv7: A Lightweight Small Object Detection Algorithm in Forest Fire Detection

Conflagration-YOLO: a lightweight object detection architecture for conflagration

LUFFD-YOLO: A Lightweight Model for UAV Remote Sensing Forest Fire Detection Based on Attention Mechanism and Multi-Level Feature Fusion

YOLOv5-CSF: an improved deep convolutional neural network for flame detection

Fs-yolo: fire-smoke detection based on improved YOLOv7

A lightweight fire detection algorithm for small targets based on YOLOv5s

Forest Fire Detection and Recognition Using YOLOv8 Algorithms from UAVs Images

Forest Fire Smoke Detection Based on Deep Learning Approaches and Unmanned Aerial Vehicle Images