Real-time Personal Protective Equipment (PPE) Detection Using YOLOv4 and TensorFlow

被引:14
作者
Protik, Adban Akib [1 ]
Rafi, Amzad Hossain [1 ]
Siddique, Shahnewaz [1 ]
机构
[1] North South Univ, Dept Elect & Comp Engn, Dhaka, Bangladesh
来源
2021 IEEE REGION 10 SYMPOSIUM (TENSYMP) | 2021年
关键词
COVID-19; Personal Protective Equipment (PPE); YOLOv4; Detector; Convolutional Neural Network (CNN); TensorFlow;
D O I
10.1109/TENSYMP52854.2021.9550808
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
The major cause for the continuous spreading of the COVID-19 disease is the human to human transmission of the virus. The virus enters the human body through exposed areas such as the mouth, nose and eyes through touching, coughing, sneezing or close proximity with infected people. So it is imperative people cover specific body parts to avoid contracting the disease and stop the transmission of the virus. Personal Protective Equipment (PPE) such as face mask, face shield, gloves help in this cause. Unfortunately some people are reluctant to wear PPE. In such situations a real-time monitoring system for PPE is needed. In this study we have developed a detector which can detect in real time if people are wearing PPE or not. The detector is developed using YOLOv4 computer vision model which specially performs well in real time object detection. We have prepared a combined dataset consisting of collected images and our captured images. We have annotated our images and image augmentation is done on all the training set images. We have also converted the detector weight to TensorFlow format to check live detection performance and added features like live object count and record keeping. In this paper we propose a method using YOLOv4 model with combined dataset and other techniques to detect four classes of objects. We get promising results from the tests and mAP of the object detector is 79%.
引用
收藏
页数:6
相关论文
共 20 条
  • [1] Anka A, 2020, YOLO V4 OPTIMAL SPEE
  • [2] [Anonymous], 2020, LABELIMG 1 8 4
  • [3] [Anonymous], The MIT License
  • [4] Bochkovskiy A., 2020, PREPRINT, DOI DOI 10.48550/ARXIV.2004.10934
  • [5] Cai WD, 2018, CHIN CONTR CONF, P9500, DOI 10.23919/ChiCC.2018.8482613
  • [6] Ghiasi G, 2018, ADV NEUR IN, V31
  • [7] Hao-Yu Lu, 2020, 2020 IEEE Eurasia Conference on IOT, Communication and Engineering (ECICE), P195, DOI 10.1109/ECICE50847.2020.9301917
  • [8] COVID-19 Detection Through Transfer Learning Using Multimodal Imaging Data
    Horry, Michael J.
    Chakraborty, Subrata
    Paul, Manoranjan
    Ulhaq, Anwaar
    Pradhan, Biswajeet
    Saha, Manas
    Shukla, Nagesh
    [J]. IEEE ACCESS, 2020, 8 : 149808 - 149824
  • [9] Hung V., 2021, YOLOV4 YOLOV3 YOLO T
  • [10] Joshua, 2020, PROCEEDINGS OF 2020 INTERNATIONAL CONFERENCE ON INFORMATION MANAGEMENT AND TECHNOLOGY (ICIMTECH), P71, DOI [10.1109/icimtech50083.2020.9211173, 10.1109/ICIMTech50083.2020.9211173]