Skip to content

This repository contains the replication package for our research project 'A Comparative Analysis of Machine Learning Approaches for Automated Face Mask Detection During COVID-19'

Notifications You must be signed in to change notification settings

al-alamin/FaceMaskDetection

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 

Repository files navigation

FaceMaskDetection: A Comparative Analysis of Machine Learning Approaches for Automated Face Mask Detection During COVID-19

Links:

Report: https://arxiv.org/abs/2112.07913 (Pre-print)

Presentation: FaceMask_detection_presentation.pptx (https://github.com/al-alamin/FaceMaskDetection/files/8617642/FaceMask_detection_presentation.pptx)


Dataset:

  • Simulated Facemask dataset (SFMD) (https://github.com/cabani/MaskedFace-Net):
    The dataset is published by Cabani et al. It contains high resolution (1024×1024) 67K simulated samples of people wearing mask correctly and incorrectly. The original images are collected from Flicker dataset. For our study, we randomly selected 6,442 masked images from SFMD and 6,442 unmasked images from the Flicker dataset.

  • Real Facemask dataset (RFMD) (https://github.com/X-zhangyang/Real-World-Masked-Face-Dataset):
    This is the biggest dataset of masked face for this problem. It consists of 90K images of 525 different people without masks and 6,442 images of real masked people. Among the masked images, many are from the same 525 people and some are different. To keep a balanced dataset, we used 6,442 masked and 6,442 unmasked images.



Machine Learning Model Experiment:

  • VGG19
  • VGG19 with Transfer Learning
  • ResNet50
  • ResNet50 with Transfer Learning
  • Cross Platform Domain Adaptation Analysis

Please read our pre-print paper for more details.

Cluster: All the experiments are conducted in the Univeristy of Calgary ARC and TALC Cluster with NVIDIA Tesla T4 GPU.

About

This repository contains the replication package for our research project 'A Comparative Analysis of Machine Learning Approaches for Automated Face Mask Detection During COVID-19'

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 99.8%
  • Python 0.2%