Research Article Open Access

Evaluation of Transfer Learning for Mask Detection

Zahin Akram1, Arifuzzaman Arman1, Mohammad Rakib Imtiaz1 and Syed Athar Bin Amir1
  • 1 North South University, Bangladesh

Abstract

The use of masks has become crucial in combating the Coronavirus pandemic. Unfortunately, the regulation of wearing a mask is not being upheld by many citizens which is contributing to the spread of the disease. To aid the efforts of regulations and to maintain safety in public areas, both large like parks or small like public transport, Artificial Intelligent systems can play a vital role. In this article, we explore the use of transfer learning across 5 models (Mobile Net V2, InceptionV3, Resnet50V2, VGG16 and DenseNet121) and measure their effectiveness in mask detection. Due to the lack of a large, diverse and annotated dataset, we explore the use of transfer learning using supervised methods and present the results of the experiments upon the Keras open-sourced models. We find an average of 99% accuracy for all 5 models. However, when we use K-Fold Cross Validation to account for bias, we find significant differences in results with the highest accuracy being achieved by VGG16 at 98.6%. With the mixture of the standard method of training and testing alongside K-Fold Cross Validation, we present our findings for the use of transfer learning for mask detection.

Journal of Computer Science
Volume 18 No. 2, 2022, 78-89

DOI: https://doi.org/10.3844/jcssp.2022.78.89

Submitted On: 16 October 2021 Published On: 3 March 2022

How to Cite: Akram, Z., Arman, A., Imtiaz, M. R. & Amir, S. A. B. (2022). Evaluation of Transfer Learning for Mask Detection. Journal of Computer Science, 18(2), 78-89. https://doi.org/10.3844/jcssp.2022.78.89

  • 2,418 Views
  • 1,000 Downloads
  • 0 Citations

Download

Keywords

  • Transfer Learning
  • Mask-Detection
  • Computer Vision
  • Object Classification
  • Covid-19