A YOVO5 Based Real-time Helmet and Mask Detection System

ABSTRACT

The COVID-19 pandemic has brought unimaginable damage to the globe;It had brought people we love away and forced the government to lock down the cities to prevent the infection of COVID-19 from spreading. This stopped various industries from working, especially the engineering and construction industries. Fortunately, the effectiveness of the COVID-19 vaccine had enabled the industries to resume their operation back to normal. However, the mask now is an essential equipment to be worn by all workers while on site, they also required to wear helmet for safety reasons. Therefore, the aim of research is to detect the helmets and masks worn by workers, if there the workers were found not for not wearing the mask properly an alert will be triggered. Five classes were determined namely 'Head,’ 'Helmet,’ 'Incorrect Mask,’ 'No Wearing Mask', and 'Wearing Mask'. A total of 1711 images of construction workers scenes were collected, and augmentation was applied on these images to generate 4733 images. All images were annotated corresponding to the defined classes. The experiments have split the training and validation dataset into a ratio of 91. The result obtain a 65.235% Mean Average Precision (mAP) as a result. © 2022, Success Culture Press. All rights reserved.