face-mask-detection
Face Mask Detection is a project to determine whether someone is wearing mask or not, using deep neural network. It contains 3 scripts that provides training new model, testing model with specific images and live face mask detection.
The gif below shows how live face mask detection works.
Model: "sequential"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
rescaling (Rescaling) (None, 180, 180, 3) 0
_________________________________________________________________
conv2d (Conv2D) (None, 180, 180, 16) 448
_________________________________________________________________
max_pooling2d (MaxPooling2D) (None, 90, 90, 16) 0
_________________________________________________________________
conv2d_1 (Conv2D) (None, 90, 90, 32) 4640
_________________________________________________________________
max_pooling2d_1 (MaxPooling2 (None, 45, 45, 32) 0
_________________________________________________________________
conv2d_2 (Conv2D) (None, 45, 45, 64) 18496
_________________________________________________________________
max_pooling2d_2 (MaxPooling2 (None, 22, 22, 64) 0
_________________________________________________________________
flatten (Flatten) (None, 30976) 0
_________________________________________________________________
dense (Dense) (None, 128) 3965056
_________________________________________________________________
dense_1 (Dense) (None, 2) 258
=================================================================
Total params: 3,988,898
Trainable params: 3,988,898
Non-trainable params: 0
_________________________________________________________________
This model has reached fantastic result of 0.9990 accuracy with test dataset( 20% ). moreover, it only takes 10 epochs to reach this result. As the graph below illustrates, from the very first epoch, accuracy is more than 0.98 and by 10th epoch, loss roughly touchs 10e-7 that is a great result.
usage
Currently, there is not any argument parser available, so if you want change any default variable, you can change variables in configurations section at the begining of each script. schema is given below.
DATASET_PATH = "./dataset/" # to provide dataset, create a directory containing all classes as a directory
# and put images to relevant class. ex:
# dataset --> no_mask --> img1, img2, ...
# correct_mask --> img1, img2, ...
OUTPUT_MODEL_PATH = 'saved_model/my_model' # output file to save model
OUTPUT_WEIGHTS_PATH = 'saved_weight/my_checkpoint2' # output file to save model's weights
EPOCHS = 10 # number of epochs
BATCH_SIZE = 32 # bach size
IMG_HEIGHT = 180 # image height
IMG_WIDTH = 180 # image width
To train a new model, after changing the configuration above, just run:
python train_model.py
To test the trained model with some specific images, run:
python test_model.py img1.jpg img2.jpg ....
At last, to run live detection mode, run:
python live_detection.py
dataset
The dataset we have used, is the combination of Flickr-Faces-HQ (FFHQ) dataset and MaskedFace-Net dataset. 5k images from each dataset. (summation of both datasets is more than 180k images)
52 Nov 9, 2022
1 Jan 13, 2022
1 Jan 1, 2022
212 Dec 29, 2022
2 Oct 10, 2022
3 Jan 6, 2022
5 Jul 18, 2022
3 Oct 2, 2021
40 Dec 22, 2022
9 Jun 16, 2022
251 Jan 05, 2023
9 Sep 26, 2022
50 Dec 03, 2022
1.7k Dec 27, 2022
105 Dec 26, 2022
103 Jan 05, 2023
46 Nov 20, 2022
0 Dec 21, 2021
141 Jan 07, 2023
9 Jul 17, 2022
1.2k Jan 02, 2023
19 Nov 05, 2022
22 Sep 15, 2022
44 Dec 14, 2022
55 Nov 01, 2022
46 Dec 12, 2022
253 Dec 27, 2022
3.1k Jan 02, 2023
226 Jan 07, 2023