基于yolov5的行人检测跟踪与社交距离预测 (pedestrian detection and social distance prediction)
生活随笔
收集整理的這篇文章主要介紹了
基于yolov5的行人检测跟踪与社交距离预测 (pedestrian detection and social distance prediction)
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
代碼、模型和示例DEMO下載地址: 下載地址
基本功能:通過距離分類人群的高危險和低危險距離。
# 課程設(shè)計
# Social Distancing based on YOLOv5.
環(huán)境要求:
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
預(yù)測demo:
python detect.py --source ./inference/videos/person.mp4
detect.py代碼:
import argparsefrom utils.datasets import *
from utils.utils import *def detect(save_img=False):out, source, weights, view_img, save_txt, imgsz = \opt.output, opt.sou
總結(jié)
以上是生活随笔為你收集整理的基于yolov5的行人检测跟踪与社交距离预测 (pedestrian detection and social distance prediction)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 原创:林彪当权后是如何对待老搭档罗荣桓的
- 下一篇: 基于yolov5和deepsort的行人