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.
python3 -m pip install -r requirements.txt
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