Enhancing face mask detection performance with comprehensive dataset and YOLOv8
10.11591/ijai.v14.i4.pp2634-2645
Trong Thua Huynh
,
Hoang Thanh Nguyen
In the context of the COVID-19 pandemic and the risk of similar infectious diseases, monitoring and promoting public health measures like wearing face masks have become crucial in controlling virus transmission. Deep learning-based mask recognition systems play an important role, but their effectiveness depends on the quality and diversity of training datasets. This study proposes the diverse and robust dataset for face mask detection (DRFMD), designed to address limitations of existing datasets and enhance mask recognition models' performance. DRFMD integrates data from sources such as AIZOO, face mask detector by Karan-Malik (KFMD), masked faces (MAFA), MOXA3K, properly wearing masked face detection dataset (PWMFD), and the Zalo AI challenge 2022, comprising 14,727 images with 29,846 instances, divided into training, validation, and testing sets. The dataset's scale and diversity ensure higher accuracy and better generalization for mask recognition models. Experiments with variations of the YOLOv8 model (n, s, m, l, x), an advanced object detection algorithm, on the DRFMD dataset, demonstrate superior performance through metrics like precision, recall, and mAP@50. Additionally, comparisons with previous dataset like FMMD show that models trained on DRFMD maintain strong generalization capabilities and higher performance. This study significantly contributes to improving accuracy of public health monitoring systems, aiding in the prevention of hazards from infectious diseases and air pollution.
