Affiliation:
1. School of Mechatronics and Information Engineering, China University of Mining and Technology (Beijing), Beijing 100083, China
2. School of Mechanical Engineering, Beijing Information Science and Technology University, Beijing 100192, China
Abstract
Flame recognition is an important technique in firefighting, but existing image flame-detection methods are slow, low in accuracy, and cannot accurately identify small flame areas. Current detection technology struggles to satisfy the real-time detection requirements of firefighting drones at fire scenes. To improve this situation, we developed a YOLOv5-based real-time flame-detection algorithm. This algorithm can detect flames quickly and accurately. The main improvements are: (1) The embedded coordinate attention mechanism helps the model more precisely find and detect the target of interest. (2) We advanced the detection layer for small targets to enhance the model’s associated identification ability. (3) We introduced a novel loss function, α-IoU, and improved the accuracy of the regression results. (4) We combined the model with transfer learning to improve its accuracy. The experimental results indicate that the enhanced YOLOv5′s mAP can reach 96.6%, 5.4% higher than the original. The model needed 0.0177 s to identify a single image, demonstrating its efficiency. In summary, the enhanced YOLOv5 network model’s overall efficiency is superior to that of the original algorithm and existing mainstream identification approaches.
Funder
High-Quality Project of the Ministry of Industry and Information Technology
Scientific Research Project of the Beijing Municipal Education Commission
Subject
Earth and Planetary Sciences (miscellaneous),Safety Research,Environmental Science (miscellaneous),Safety, Risk, Reliability and Quality,Building and Construction,Forestry
Cited by
2 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献