Windows 10:
Anaconda 3
python 3.6.17
GPU:
Nvidia GTX 940MX
Algorithm for calculating number of people wearing mask and not wearing mask.
Download yolov5 weights from here
Download DeepSort weights from here and place ckpt.t7 file under pytorch_deep_sort/deep_sort/deep/checkpoint/
git clone https://github.com/HarshRangwala/DeepSort-Surveillance
cd DeepSort-Surveillance
pip3 install -r requirements.txt
To run the main algorithm. Execute the command:
python PersonCounter.py --source <VIDEO_FILENAME> --device 0 --weights yolov5/weights/<WEIGHT_FILENAME>