基于yolov5的行人检测跟踪与社交距离预测 （pedestrian detection and social distance prediction）

Posted 2023-01-02 踟蹰横渡口,彳亍上滩舟。

 代码、模型和示例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代码：

import argparse

from utils.datasets import *
from utils.utils import *


def detect(save_img=False):
    out, source, weights, view_img, save_txt, imgsz = \\
        opt.output, opt.sou