Covid-negative but positive in loving you <3.
- Require Python 3.6+
- Clone this repository
- Go to src folder and run
pip install -r requirements.txt
-
Go to src/distance folder and run
python distance.py --video path
First frame will be shown, then you have to click to choose 6 points:
- First 4 points are the top-left, top-right, bottom-left, bottom-right corners of the ROI.
- The last 2 points define distance in horizontal and vertical, which equals to 1 meter in the real world.
The result will to save to distance.txt.
-
Go to src folder and download pretrained weights for YOLOv5, make sure pt files in weights folder.
-
Go to src folder and run
bash demo.sh <video path> <limit number of people>
The result will be saved in
src\runs\track\test\<video name>
| Google Colab demo |
|---|
 |
First, cam on giang vien.
Second, allmost our code is borrowed from these resources:
- YOLOv5 and deepakcrk's pretrained weights for head & person detection.
- ByteTrack for tracking method.
Many thanks for their wonderful work.
Finally, team spirit is the most important factor to win victory. Thank you for all your hảd wỏk. We did very well :>.
- Person and head detection
- Add tracking method to improve detection performance
- Face mask classifier
- Top-view localization and distance estimation
- Optimize the face mask classifier
- Train more yolov5 models at more scale