A Flame-Detection Algorithm Using the Improved YOLOv5
FIRE-SWITZERLAND(2023)
摘要
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, a-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 & PRIME;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.
更多查看译文
关键词
flame recognition,artificial intelligence,real-time detection,YOLOv5,boundary loss function
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要