shahizat / Face-Mask-Detection

Real time face-mask detection system using Deep Learning and TensorRT

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

Technologies

• PyTorch
• SSD Mobilenet
• TensorRT
• Python

Usage

Follow the steps provided here.

Data

The dataset used for training the model is available here.