中文文档

Gen2 COVID-19 mask detection

This experiment allows you to run the COVID-19 mask/no-mask object detector which was trained via the Google Colab tutorial here.

You can leverage this notebook and add your own training images to improve the quality of this mask detector.
Our training was a quick weekend effort at it.

Demo

Install requirements

python3 -m pip install -r requirements.txt

Usage

usage: main.py [-h] [-nd] [-cam] [-vid VIDEO]

optional arguments:
  -h, --help            show this help message and exit
  -nd, --no-debug       Prevent debug output
  -cam, --camera        Use DepthAI 4K RGB camera for inference (conflicts with -vid)
  -vid VIDEO, --video VIDEO
                        Path to video file to be used for inference (conflicts with -cam)

To use with a video file, run the script with the following arguments

python3 main.py -vid ./input.mp4

To use with DepthAI 4K RGB camera, use instead

python3 main.py -cam