Return
A Flame-Detection Algorithm Using the Improved YOLOv5
DOI:10.3390/fire6080313.png)
Abstract
En 中文
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.
Keywords:
flame recognition
artificial intelligence
real-time detection
YOLOv5
boundary loss function
Journal
F
IF:
2.7
Papers:
1.7K
Citations:
3.2K
Organization
No organization information available
Cited Papers
SSP-Net: Scalable sequential pyramid networks for real-Time 3D human pose regression
PATTERN RECOGNITION
IF7.6
Early fire detection using convolutional neural networks during surveillance for effective disaster management
NEUROCOMPUTING
IF6.5
Comparative study of layer by layer assembled multilayer films based on graphene oxide and reduced graphene oxide on flexible polyurethane foam: flame retardant and smoke suppression properties
RSC Advances
IF0

