Face Mask Detection System using AI and NVIDIA Jetson board

This repo implements SSD (Single Shot MultiBox Detector) in PyTorch for face mask detection, using MobileNet backbones.

For documentation, please refer to Object Detection portion of the Hello AI World tutorial: Re-training SSD-Mobilenet

Demo

Follow the steps provided here.

The dataset used for training the model is available here.

Name		Name	Last commit message	Last commit date
Latest commit History 17 Commits
data/MaskDetector		data/MaskDetector
models/mymodel		models/mymodel
vision/utils		vision/utils
1.jpg		1.jpg
2.jpg		2.jpg
LICENSE		LICENSE
README.md		README.md
eval_ssd.py		eval_ssd.py
onnx_export.py		onnx_export.py
open_images_classes.txt		open_images_classes.txt
open_images_downloader.py		open_images_downloader.py
output.jpg		output.jpg
output.mp4		output.mp4
prepare.py		prepare.py
requirements.txt		requirements.txt
run_ssd_example.py		run_ssd_example.py
train_ssd.py		train_ssd.py
video.mp4		video.mp4