China Safety Science Journal ›› 2024, Vol. 34 ›› Issue (5): 155-161.doi: 10.16265/j.cnki.issn1003-3033.2024.05.1050

• Safety engineering technology • Previous Articles     Next Articles

Firework detection method based on improved YOLO-V5 algorithm

ZHANG Mingzhen1(), DUAN Jiangzhong1, LIANG Zhaowei2, GUO Junjie1, CHAI Dashan3   

  1. 1 Shenzhen Urban Public Safety and Technology Institute,Shenzhen Guangdong 518038, China
    2 School of Big Data and Internet, Shenzhen Technology University, Shenzhen Guangdong 518118, China
    3 Shenzhen Branch of China Tower, Shenzhen Guangdong 518000, China
  • Received:2023-11-25 Revised:2024-02-26 Online:2024-05-28 Published:2024-11-28

Abstract:

To reduce the influences of background interference factors in natural environments such as clouds, mist, dust, lights, sunrise, and sunset on the smoke and flame target detection accuracy, a smoke and fire detection algorithm based on an improved YOLO-V5 algorithm was proposed. Smoke, flame target images, and interference image data sets were obtained from the on-site collection and web crawling approaches to solve sample imbalance and improve model generalization ability. A bidirectional feature pyramid network (BiFPN) was used to replace the original feature pyramid network (FPN) + path aggregation network (PAN) structure, and then multi-scale feature fusion on the target was performed to strengthen the model feature fusion ability. At the same time, distance intersection-over-union(DIoU) non-maximum suppression(NMS) is used to replace the original non-maximum suppression (NMS) to speed up the convergence of the detection box loss function and enhance the model reasoning ability. The results showed that the improved algorithm's accuracy, recall rate, mean average precision(mAP) and FPR were 79.2%, 68.6%, 74.2%, and 12.8%, respectively. Compared with the original YOLO-V5 algorithm, the proposed algorithm improved accuracy rate, recall rate, and mAP by 1.9%, 0.9%, and 2.7%, respectively. Furthermore, the FPR was decreased by 3.7%.

Key words: YOLO-V5 algorithm, smoke, fire, target detection, false positive rate(FPR)

CLC Number: