Face Mask Type Detector

This repository contains the code and Google Colab notebook for training a face mask type detector with TensorFlow Object Detection API.

The blog post of this project can be found on Medium.

The model is used to detect face masks of 4 classes: homemade cloth covers, surgical mask, n95 mask and no mask. The corresponding labels are:

• homemade
• surgical 
• n95
• bare

Dependencies on Windows 10

• Python 3.5
• TensorFlow 1.10

Notebook used for training on Google Colab

Training image examples

The images used for training are from Google and Upslashe. There are 247, 197, 184 and 255 images for each class.

• homeamde

• surgical 

• n95

• bare

Model Output

• Testing on images from the Internet

• Testing with a friend in low light. All masks were labelled correctly!

• Test images from a Raspberry Pi camera