Abstract:For fire detection, there are characteristics such as variable samples feature morphology, complex background and dense targets, small samples size of dataset and imbalance of categories, which lead to the problems of low accuracy and poor real-time performance of the existing fire detection models. We propose EA-YOLO, a flame and smoke detection model based on efficient multi-scale feature enhancement. In order to improve the extraction capability of the network model for flame smoke targets' features, an efficient attention mechanism is integrated into the backbone network, Multi Channel Attention (MCA), and the number of parameters of the model is reduced by introducing the RepVB module; at the same time, we design a multi-weighted, multidirectional feature neck structure called the Multidirectional Feature Pyramid Network (MDFPN), to enhance the model's flame smoke target feature information fusion ability; finally, we redesign the CIoU loss function by introducing the Slide weighting function to improve the imbalance between difficult and easy samples. Additionally, to address the issue of small sample sizes in fire datasets, we establish two new fire datasets: Fire-smoke and Ro-fire-smoke. The latter includes a model robustness validation function. The experimental results show that the method of this paper is 6.5 and 7.3 higher compared to the baseline model YOLOv7 on the Fire-smoke and Ro-fire-smoke datasets, respectively. The detection speed is 74.6 frames per second. To fully demonstrate the superiority of EA-YOLO, we utilized the public FASDD dataset and compared several state-of-the-art (SOTA) models with EA-YOLO on this dataset. The results were highly favorable. It fully demonstrates that the method in this paper has high fire detection accuracy while considering the real-time nature of the model. The source code and datasets are located at https://github.com/DIADH/DIADH.YOLO.

YOLOv8 for Fire and Smoke Recognition Algorithm Integrated with the Convolutional Block Attention Module

Research on Fire Smoke Detection Algorithm Based on Improved YOLOv8

Light-YOLOv5: A Lightweight Algorithm for Improved YOLOv5 in Complex Fire Scenarios

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

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

Industrial and Mining Fire Detection Algorithm Based on Improved YOLO

FS-YOLO: Real-time Fire and Smoke Detection Based on Improved Object Detection Algorithms

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

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

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

Fire and smoke real-time detection algorithm for coal mines based on improved YOLOv8s

An Improved YOLOv5s Fire Detection Model

A Smoke Detection Model Based on Improved YOLOv5

GPAC-YOLOv8: Lightweight Target detection for fire scenarios

Lightweight Forest Smoke and Fire Detection Algorithm Based on Improved YOLOv5.

Early Smoke Detection Based on Improved YOLO-PCA Network

Real‐time fire and smoke detection with transfer learning based on cloud‐edge collaborative architecture

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

Real-time Detection of Flame and Smoke Using an Improved YOLOv4 Network

An Improved Fire and Smoke Detection Method Based on YOLOv8n for Smart Factories

A forest fire detection method based on improved YOLOv5