Skip to content

A face mask detector model to classify if a face image wears a mask correctly, incorrectly or none at all

License

Notifications You must be signed in to change notification settings

Eserichard/facemask-detector_v1.0

Repository files navigation

facemask-detector

Data Download Scripts: To download the images from one drive and google drive using CURL.

DataProcessing.py: To split the downloaded images into train-validation-holdout and organize the files for ImageFolder.

MaskDetector_Final.py: Training of the final resnet50 model.

Inference.py: To view the inference probability of each class based on input image passed.

OcclusionExperiment.py: To intrepret and visualize the CNN based on occlusion experiment

About

A face mask detector model to classify if a face image wears a mask correctly, incorrectly or none at all

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •  

Languages