代码、模型和示例DEMO下载地址: 下载地址
基本功能:通过距离分类人群的高危险和低危险距离。
# 课程设计
# Social Distancing based on YOLOv5.
环境要求:
Cython
numpy==1.17
opencv-python
torch>=1.5
matplotlib
pillow
tensorboard
PyYAML>=5.3
torchvision
scipy
tqdm
git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI
预测demo:
python detect.py --source ./inference/videos/person.mp4
detect.py代码: