20 December 2024 Factory smoke and fire electrical target detection model based on improved YOLOv5
Zenghua Li, Qingsong Han, Xiufang Yan, Libo Bai, Xu Wang, Fan Yang, Jie Li
Author Affiliations +
Abstract

Real-time and accurate detection of flames, smoke, and electric arcs is an important prerequisite for the safe production in factories. When dealing with complex features of flames, smoke, and arc targets, traditional detection algorithms suffer from insufficient accuracy, high missed detection rates for small targets. We proposed a smoke and fire electrical detection model MS-YOLOv5 based on YOLOv5s. First, the model’s detection accuracy for small targets is improved by clustering the anchor frame sizes suitable for small target detection based on the K-mean++ clustering algorithm. Second, the C3-DySnake module was designed, and dynamic snake convolution was utilized to replace the normal convolution module in the C3 module of the backbone network, giving the backbone network the flexibility to adaptively focus on key features. Third, the multidimensional collaborative attention module was added to the end of the neck network, so that the model focused on the local area information in the feature fusion stage while obtaining a stronger feature map. Finally, the alpha-intersection over union (α-IoU) loss function was used to retain the high-quality prediction frames, and effectively improve the model convergence speed. The MS-YOLOv5 detection model established on a self-constructed smoke and fire electrical dataset was high in precision and effective in recall, with better mAP@0.5-value compared with the original YOLOv5.

© 2024 SPIE and IS&T

Funding Statement

Zenghua Li, Qingsong Han, Xiufang Yan, Libo Bai, Xu Wang, Fan Yang, and Jie Li "Factory smoke and fire electrical target detection model based on improved YOLOv5," Journal of Electronic Imaging 33(6), 063050 (20 December 2024). https://doi.org/10.1117/1.JEI.33.6.063050
Received: 30 June 2024; Accepted: 22 November 2024; Published: 20 December 2024
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Fire

Target detection

Convolution

Education and training

Performance modeling

Detection and tracking algorithms

Data modeling

RELATED CONTENT


Back to Top