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.

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

Deep Learning Based Fire Detection System For Surveillance Videos

Flame Detection Using Deep Learning

Multi-scale Forest Flame Detection Based on Improved and Optimized YOLOv5

An Improved YOLOv5s Fire Detection Model

Multi-Scale Video Flame Detection for Early Fire Warning Based on Deep Learning

EFA-YOLO: An Efficient Feature Attention Model for Fire and Flame Detection

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

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

Conflagration-YOLO: a lightweight object detection architecture for conflagration

An Improved Lightweight YOLOv8 Network for Early Small Flame Target Detection

DCNFYOLO: Dual-Convolution Network and Feature Fusion for High-Precision Smoke Detection

Multi-Scale Forest Fire Recognition Model Based on Improved YOLOv5s

Ea-yolo: efficient extraction and aggregation mechanism of YOLO for fire detection

Fire Safety Detection Based on CAGSA-YOLO Network

A Two-Stream CNN Model with Adaptive Adjustment of Receptive Field Dedicated to Flame Region Detection

DG-YOLO: A Novel Efficient Early Fire Detection Algorithm Under Complex Scenarios

A Small Target Forest Fire Detection Model Based on YOLOv5 Improvement

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

Industrial and Mining Fire Detection Algorithm Based on Improved YOLO

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