当前位置: 代码迷 >> 综合 >> pytorch-yolov5_deepsort目标跟踪行人车辆计数
  详细解决方案

pytorch-yolov5_deepsort目标跟踪行人车辆计数

热度:72   发布时间:2023-10-19 18:45:46.0

pytorch-yolov5_deepsort目标跟踪行人车辆计数

 看图就可以知道。穿过两条线,分别用于测上行人流和下行人流。

这个cpu和gpu版本都可以跑,cpu的话,安装好相应的库之后,运行会报

RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘

的错误,原因是模型是利用fp16混合精度计算对CPU进行推理,不安装gpu版本的话是不支持fp16的,因此需要将代码中的half.()修改成.float()即可解决问题。

回到代码部分

项目是github上某大佬开源的

整个安装和操作流程是非常的简便

如果是没有python环境的,可以看我之前上传的博客,有关于python的安装教程和IDE编译器安装

带你快速上手

python-anaconda安装

在win10上安装pytorch-gpu版本</