arrow
Return

A Flame-Detection Algorithm Using the Improved YOLOv5

delete2023-08-13
delete5
delete
OA
AI
X
Xingang Xie
陈
陈珂 (Ke Chen)
Y
Yiran Guo
T
Tan, Botao
L
Lumeng Chen
M
Min Huang *
DOI:10.3390/fire6080313delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

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
Fire Switzerland
IF:
2.7
Papers:
1.7K
Citations:
3.2K

Organization

No organization information available
Cited Papers

Cited Papers

errShare
errSave
Convolutional Neural Networks Based Fire Detection in Surveillance Videos
err2018-01-01
err320
errOAAI
errMuhammad, Khan; Ahmad, Jamil; Mehmood, Irfan; Rho, Seungmin; Baik, Sung Wook
errShare
errSave
SSP-Net: Scalable sequential pyramid networks for real-Time 3D human pose regression
err2023-10-01
err7
errOAAI
errLuvizon, Diogo Carbonera; Tabia, Hedi; Picard, David
errShare
errSave
Enhancing Geometric Factors in Model Learning and Inference for Object Detection and Instance Segmentation
err2022-08-01
err620
errOAAI
errZheng, Zhaohui; Wang, Ping; Ren, Dongwei; Liu, Wei; Ye, Rongguang; Hu, Qinghua; Zuo, Wangmeng
errShare
errSave
errShare
errSave
Anchor-Free Smoke and Flame Recognition Algorithm with Multi-Loss
err2023-06-04
err1
errOAAI
errLi, Gang; Chen, Peng; Xu, Chuanyun; Sun, Chengjie; Ma, Yingli
errShare
errSave
Fast fire flame detection in surveillance video using logistic regression and temporal smoothing
err2016-01-01
err101
PREAI
errKong, Seong G.; Jin, Donglin; Li, Shengzhe; Kim, Hakil
errShare
errSave
researcher View more