Skip to main content
SpringerLink
Log in

YOLOv4 algorithm for the real-time detection of fire and personal protective equipments at construction sites

  • 1200: Machine Vision Theory and Applications for Cyber Physical Systems
  • Published:
Multimedia Tools and Applications Aims and scope Submit manuscript

Abstract

Many difficulties are encountered during evacuation from construction sites in hazardous situations, which may lead to severe fatalities. These fatalities, especially caused by fire, may be significantly reduced by ensuring personal protective equipment (PPE) compliance of construction site workers and fire detection through proper surveillance. Thus, the detection of PPEs, fire and injured or trapped persons, can greatly assist in the reduction of fatalities and economic loss. This article presents a novel approach towards the detection of fire and PPEs to assist in the monitoring and evacuation tasks. This work utilizes the YOLOv4 and YOLOv4-tiny algorithms based on deep learning for carrying out the detection task. A self-made dataset has been utilized to train the model in the Darknet neural network framework. Moreover, a comparative analysis with previous works has been carried out in order to endorse the real-time efficacy of the proposed work. The results verify the strength of YOLOv4 algorithm in real-time detection and surveillance at construction sites with maximum mean average precision (mAP) of 76.86 %.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Log in via an institution

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7

Similar content being viewed by others

Abbreviations

AP:

Average Precision

CNN:

Convolutional Neural Networks

DL:

Deep Learning

IoU:

Intersection over Union

mAP:

Mean Average Precision

PPE:

Personal Protective Equipment

RFID:

Radio Frequency Identification

YOLO:

You Only Look Once

References

  1. Akbar-Khanzadeh F (1998) Factors contributing to discomfort or dissatisfaction as a result of wearing personal protective equipment. J Hum Ergol (Tokyo) 27:70–75

    Google Scholar 

  2. Balakreshnan B, Richards G, Nanda G et al (2020) PPE Compliance Detection using Artificial Intelligence in Learning Factories. Procedia Manuf 45:277–282. https://doi.org/10.1016/j.promfg.2020.04.017

  3. Barro-Torres S, Fernández-Caramés TM, Pérez-Iglesias HJ, Escudero CJ (2012) Real-time personal protective equipment monitoring system. Comput Commun 36:42–50. https://doi.org/10.1016/j.comcom.2012.01.005

  4. Bhole SA (2016) Safety Problems and Injuries on Construction Site: A Review. Int J Eng Tech 2:24–35

    Google Scholar 

  5. Bochkovskiy A, Wang C-Y, Liao H-YM (2020) YOLOv4: Optimal Speed and Accuracy of Object Detection. arXiv preprint arXiv:2004.10934

  6. Chen RC (2019) Automatic License Plate Recognition via sliding-window darknet-YOLO deep learning. Image Vis Comput 87:47–56. https://doi.org/10.1016/j.imavis.2019.04.007

    Article  Google Scholar 

  7. Ding L, Fang W, Luo H et al (2018) A deep hybrid learning model to detect unsafe behavior: Integrating convolution neural networks and long short-term memory. Autom Constr 86:118–124. https://doi.org/10.1016/j.autcon.2017.11.002

    Article  Google Scholar 

  8. Dundar A, Jin J, Martini B, Culurciello E (2017) Embedded streaming deep neural networks accelerator with applications. IEEE Trans Neural Netw Learn Syst 28:1572–1583. https://doi.org/10.1109/TNNLS.2016.2545298

    Article  MathSciNet  Google Scholar 

  9. Fang Q, Li H, Luo X et al (2018) Detecting non-hardhat-use by a deep learning method from far-field surveillance videos. Autom Constr 85:1–9. https://doi.org/10.1016/j.autcon.2017.09.018

  10. Hassaballah M, Awad AI (2020) Deep Learning in Computer Vision. CRC Press

    Book  Google Scholar 

  11. Karthik R, Hariharan M, Anand S et al (2020) Attention embedded residual CNN for disease detection in tomato leaves. Appl Soft Comput 86:105933. https://doi.org/10.1016/j.asoc.2019.105933

  12. Kelm A, Laußat L, Meins-Becker A et al (2013) Mobile passive Radio Frequency Identification (RFID) portal for automated and rapid control of Personal Protective Equipment (PPE) on construction sites. Autom Constr 36:38–52. https://doi.org/10.1016/j.autcon.2013.08.009

  13. Kolar Z, Chen H, Luo X (2018) Transfer learning and deep convolutional neural networks for safety guardrail detection in 2D images. Autom Constr 89:58–70. https://doi.org/10.1016/j.autcon.2018.01.003

    Article  Google Scholar 

  14. Kumar S, Yadav D, Gupta H et al (2020) A Novel YOLOv3 Algorithm-Based Deep Learning Approach for Waste Segregation: Towards Smart Waste Management. Electronics 10:14. https://doi.org/10.3390/electronics10010014

    Article  Google Scholar 

  15. Lee D-H (2020) CNN-based single object detection and tracking in videos and its application to drone detection. Multimed Tools Appl. https://doi.org/10.1007/s11042-020-09924-0

    Article  Google Scholar 

  16. Li P, Zhao W (2020) Image fire detection algorithms based on convolutional neural networks. Case Stud Therm Eng 19. https://doi.org/10.1016/j.csite.2020.100625

  17. Luo Y, Zhao L, Liu P, Huang D (2018) Fire smoke detection algorithm based on motion characteristic and convolutional neural networks. Multimed Tools Appl 77:15075–15092. https://doi.org/10.1007/s11042-017-5090-2

    Article  Google Scholar 

  18. Man-Woo P, Nehad E, Zhenhua Z (2015) Hardhat-Wearing Detection for Enhancing On-Site Safety of Construction Workers. J Constr Eng Manag 141:4015024. https://doi.org/10.1061/(ASCE)CO.1943-7862.0000974

    Article  Google Scholar 

  19. Mao W, Wang W, Dou Z, Li Y (2018) Fire Recognition Based On Multi-Channel Convolutional Neural Network. Fire Technol 54:531–554. https://doi.org/10.1007/s10694-017-0695-6

    Article  Google Scholar 

  20. Mneymneh BE, Abbas M, Khoury H (2017) Automated Hardhat Detection for Construction Safety Applications. Procedia Eng 196:895–902. https://doi.org/10.1016/j.proeng.2017.08.022

  21. Mneymneh BE, Abbas M, Khoury H (2019) Vision-Based Framework for Intelligent Monitoring of Hardhat Wearing on Construction Sites. J Comput Civ Eng 33:1–20. https://doi.org/10.1061/(ASCE)CP.1943-5487.0000813

    Article  Google Scholar 

  22. Muhammad K, Ahmad J, Mehmood I et al (2018) Convolutional Neural Networks Based Fire Detection in Surveillance Videos. IEEE Access 6:18174–18183. https://doi.org/10.1109/ACCESS.2018.2812835

    Article  Google Scholar 

  23. Muhammad K, Ahmad J, Baik SW (2018) Early fire detection using convolutional neural networks during surveillance for effective disaster management. Neurocomputing 288:30–42. https://doi.org/10.1016/j.neucom.2017.04.083

    Article  Google Scholar 

  24. Muhammad K, Khan S, Baik SW (2020) Efficient Convolutional Neural Networks for Fire Detection in Surveillance Applications. https://books.google.com https://doi.org/10.1201/9781351003827-3

  25. Namozov A, Cho YI (2018) An efficient deep learning algorithm for fire and smoke detection with limited data. Adv Electr Comput Eng 18:121–128. https://doi.org/10.4316/AECE.2018.04015

    Article  Google Scholar 

  26. Naticchia B, Vaccarini M, Carbonari A (2013) A monitoring system for real-time interference control on large construction sites. Autom Constr 29:148–160. https://doi.org/10.1016/j.autcon.2012.09.016

  27. Nath ND, Chaspari T, Behzadan AH (2019) Single- And multi-label classification of construction objects using deep transfer learning methods. J Inf Technol Constr 24:511–526. https://doi.org/10.36680/J.ITCON.2019.028

  28. Nath ND, Behzadan AH, Paal SG (2020) Deep learning for site safety: Real-time detection of personal protective equipment. Autom Constr 112:103085. https://doi.org/10.1016/j.autcon.2020.103085

    Article  Google Scholar 

  29. Nie X, Yang M, Liu RW (2019) Deep Neural Network-Based Robust Ship Detection Under Different Weather Conditions. In: 2019 IEEE Intelligent Transportation Systems Conference, ITSC 2019. Inst Electr Electr Eng Inc 47–52

  30. Park M-W, Brilakis I (2012) Construction worker detection in video frames for initializing vision trackers. Autom Constr 28:15–25. https://doi.org/10.1016/j.autcon.2012.06.001

  31. Rangel JC, Martínez-Gómez J, Romero-González C et al (2018) Semi-supervised 3D object recognition through CNN labeling. Appl Soft Comput 65:603–613. https://doi.org/10.1016/j.asoc.2018.02.005

  32. Redmon J, Farhadi A (2018) YOLOv3: An Incremental Improvement. arXiv preprint arXiv:1804.02767

  33. Seo J, Han S, Lee S, Kim H (2015) Computer vision techniques for construction safety and health monitoring. Adv Eng Inform 29:239–251. https://doi.org/10.1016/j.aei.2015.02.001

    Article  Google Scholar 

  34. Seong H, Son H, Kim C (2018) A Comparative Study of Machine Learning Classification for Color-based Safety Vest Detection on Construction-Site Images. KSCE J Civ Eng 22:4254–4262. https://doi.org/10.1007/s12205-017-1730-3

    Article  Google Scholar 

  35. Siddula M, Dai F, Ye Y, Fan J (2016) Unsupervised Feature Learning for Objects of Interest Detection in Cluttered Construction Roof Site Images. Procedia Eng 145:428–435. https://doi.org/10.1016/j.proeng.2016.04.010

    Article  Google Scholar 

  36. Sun L, Zhao C, Yan Z et al (2019) A novel weakly-supervised approach for RGB-D-based nuclear waste object detection. IEEE Sens J 19:3487–3500. https://doi.org/10.1109/JSEN.2018.2888815

    Article  Google Scholar 

  37. Tran Q-H, Le T-L, Hoang S-H (2019) A fully automated vision-based system for real-time personal protective detection and monitoring. KICS Korea-Vietnam Int Jt Work Commun Inf Sci 2019:1–6

    Google Scholar 

  38. Wu J, Cai N, Chen W et al (2019) Automatic detection of hardhats worn by construction personnel: A deep learning approach and benchmark dataset. Autom Constr 106:102894. https://doi.org/10.1016/j.autcon.2019.102894

  39. Wu D, Lv S, Jiang M, Song H (2020) Using channel pruning-based YOLO v4 deep learning algorithm for the real-time and accurate detection of apple flowers in natural environments. Comput Electron Agric 178:105742. https://doi.org/10.1016/j.compag.2020.105742

  40. Yin Z, Wan B, Yuan F et al (2017) A Deep Normalization and Convolutional Neural Network for Image Smoke Detection. IEEE Access 5:18429–18438. https://doi.org/10.1109/ACCESS.2017.2747399

    Article  Google Scholar 

Download references

Author information

Authors and Affiliations

  1. Department of Electrical Engineering, Indian Institute of Technology, Roorkee, India

    Saurav Kumar

  2. Department of Instrumentation and Control Engineering, Dr. B. R. Ambedkar National Institute of Technology Jalandhar, Jalandhar, India

    Himanshu Gupta & Om Prakash Verma

  3. Faculty of Informatics, Technische Universität Wien, 1040, Vienna, Austria

    Drishti Yadav

  4. Department of Electronics and Communication Engineering, Design and Manufacturing, PDPM Indian Institute of Information Technology, Jabalpur, India

    Irshad Ahmad Ansari

Authors
  1. Saurav Kumar
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. Himanshu Gupta
    View author publications

    You can also search for this author in PubMed Google Scholar

  3. Drishti Yadav
    View author publications

    You can also search for this author in PubMed Google Scholar

  4. Irshad Ahmad Ansari
    View author publications

    You can also search for this author in PubMed Google Scholar

  5. Om Prakash Verma
    View author publications

    You can also search for this author in PubMed Google Scholar

Corresponding author

Correspondence to Om Prakash Verma.

Additional information

Publisher's Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Kumar, S., Gupta, H., Yadav, D. et al. YOLOv4 algorithm for the real-time detection of fire and personal protective equipments at construction sites. Multimed Tools Appl 81, 22163–22183 (2022). https://doi.org/10.1007/s11042-021-11280-6

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s11042-021-11280-6

Keywords

Search

Navigation