看图就可以知道。穿过两条线,分别用于测上行人流和下行人流。
这个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版本</