https://github.com/balnarendrasapa/mask-detection
Mask Detection using Deep Learning. This repository contains code to train a model that can detect whether a person is wearing mask or not.
https://github.com/balnarendrasapa/mask-detection
deep-learning mask-detection neural-network transfer-learning vgg19
Last synced: 4 months ago
JSON representation
Mask Detection using Deep Learning. This repository contains code to train a model that can detect whether a person is wearing mask or not.
- Host: GitHub
- URL: https://github.com/balnarendrasapa/mask-detection
- Owner: balnarendrasapa
- License: gpl-3.0
- Created: 2023-11-19T20:58:31.000Z (almost 2 years ago)
- Default Branch: master
- Last Pushed: 2023-11-22T20:37:10.000Z (almost 2 years ago)
- Last Synced: 2023-11-23T21:35:57.785Z (almost 2 years ago)
- Topics: deep-learning, mask-detection, neural-network, transfer-learning, vgg19
- Language: Jupyter Notebook
- Homepage:
- Size: 20.5 MB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Mask Detection
This project is about building a model that can detect if a person is wearing a mask or not. I build the model using pytorch in jupyter notebook. This project makes use of transfer-learning. I have used vgg19 pretrained model and added new layers on top of them and trained the resultant model.
## Dataset
The dataset is taken from [here](https://www.kaggle.com/datasets/ashishjangra27/face-mask-12k-images-dataset) which contains almost 12k images.
## Running the project
The project is developed in google colab as it provides a free T4-GPU with almost 15 gb of VRAM. click on the above badge to open the project in the google colab.
## Result
