An Improved Fire and Smoke Detection Method Based on YOLOv7

Abstract

Fires can cause extensive damage to human life and property, highlighting the importance of accurate flame and smoke detection systems. However, current methods for detecting flame and smoke struggle to balance the demands of real-time processing and prediction accuracy, limiting their applicability in fire warning and detecting systems. In this paper, we propose a novel approach based on the YOLOv7 architecture for efficient and accurate detection of fire and smoke in images. Our approach incorporates partial convolutional layers into the E-ELAN module of the YOLOv7 network, enabling faster and more precise identification of fire and smoke. Furthermore, we introduce the Focal-EIoU loss function to address the issue of fluctuating model loss caused by low-quality samples. We validate our approach on a real-world dataset and report significant improvements in detection accuracy, with a mean average precision of 78.5 and an FPS increase to 63. These results demonstrate the effectiveness of our approach in enhancing the capability of fire and smoke detection systems.