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.

Learning precise feature via self-attention and self-cooperation YOLOX for smoke detection

A Slight Smoke Perceptual Network.

Early Smoke Detection Based on Improved YOLO-PCA Network

A Novel Smoke Detection Algorithm Based on Improved Mixed Gaussian and YOLOv5 for Textile Workshop Environments

A Smoke Detection Model Based on Improved YOLOv5

Real-Time Fire Smoke Detection Method Combining a Self-Attention Mechanism and Radial Multi-Scale Feature Connection

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

SM-YOLO: A Model for Real-Time Smoke Detection

Smoking-YOLOv8: a novel smoking detection algorithm for chemical plant personnel

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

Video Smoke Detection Based on Deep Saliency Network

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

Early Wildfire Smoke Detection Using Different YOLO Models

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

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

Research on Fire Smoke Detection Algorithm Based on Improved YOLOv8

An Efficient and Lightweight Detection Model for Forest Smoke Recognition

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

An Improved YOLOv5s Fire Detection Model

A high-precision lightweight smoke detection model based on SE attention mechanism

Enhanced Receptive Field Smoke Detection Model Embedded with Attention Mechanism