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
- Python 3.5
- TensorFlow 1.10
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
-
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
