车辆识别——yolov5 + DeepSort_Pytorch

项目地址:https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch.git

1 直接下载项目至本地(git clone弄不下来)

2 按照文档下载其中的子工程

git init #必须先使用这个,因为我们不是git clone下载的,现在就没有git目录
git submodule update --init

3 手动导入yolov5模块

可能报错:ModuleNotFoundError: No module named 'yolov5.models'

去github找yolov5的“官方文档”,readme长下面这样,放入yolov5的目录下面

车辆识别——yolov5 + DeepSort_Pytorch_第1张图片

直接全部黏贴过来

车辆识别——yolov5 + DeepSort_Pytorch_第2张图片

 4 运行程序,命令为python track.py --source +识别的内容名

车辆识别——yolov5 + DeepSort_Pytorch_第3张图片

 错误:No module named 'torchreid'

分析:pycharm中并没有这个包,简单使用pip install 命令也无法安装

解决:https://github.com/KaiyangZhou/deep-person-reid.git

下载这个,找出里面的torchreid文件夹,直接复制黏贴到和track.py同级的目录下。

车辆识别——yolov5 + DeepSort_Pytorch_第4张图片

再次运行,将会自动安装torchreid

车辆识别——yolov5 + DeepSort_Pytorch_第5张图片

 5 该项目成功运行,可进行进一步学习

 

你可能感兴趣的:(图像/视频处理,pytorch,深度学习,人工智能)