手把手教你搭建YOLOv5 + StrongSORT with OSNet环境

举报
AI浩 发表于 2023/01/08 16:33:18 2023/01/08
【摘要】 项目介绍YOLOv5 + StrongSORT with OSNet模型由YOLOv5检测器 、StrongSORT跟踪算法、OSNet行人重识别模型组成项目地址:https://github.com/mikel-brostrom/Yolov5_StrongSORT_OSNet由于现在的代码中没有yolov5和reid代码,所以需要下载源码加入到YOLOv5 + StrongSORT wi...

项目介绍

YOLOv5 + StrongSORT with OSNet模型由YOLOv5检测器 、StrongSORT跟踪算法、OSNet行人重识别模型组成
项目地址:
https://github.com/mikel-brostrom/Yolov5_StrongSORT_OSNet
由于现在的代码中没有yolov5和reid代码,所以需要下载源码加入到YOLOv5 + StrongSORT with OSNet项目中。
yolov5下载地址:
在这里插入图片描述

https://github.com/ultralytics/yolov5/tree/91a81d48fa4e34dbdbaf0e45a1f841c11216aab5
下载代码后,放到yolov5的文件夹中,如下图:
在这里插入图片描述

在这里插入图片描述

https://github.com/KaiyangZhou/deep-person-reid
将下载的代码放到如下位置:
在这里插入图片描述

安装缺少的库

安装lap

pip install lap

安装filterpy

pip install filterpy

安装Cython-bbox
这里主要讲如何在Window下面安装:

第一步:从网址https://pypi.org/project/cython-bbox/中下载cython-bbox,然后解压,可以看到dist文件夹。
在这里插入图片描述

第二步:解压后,在文件夹里找到setup.py,并把extra_compile_args=[’-Wno-cpp’],修改为 extra_compile_args = {‘gcc’: [’/Qstd=c99’]},如下图所示:

   extra_compile_args = {'gcc':['/Qstd=c99']},

在这里插入图片描述
完成修改后,执行命令:

python setup.py build_ext install

下载权重文件

下载地址:https://github.com/BrianLangat/Yolov5_StrongSORT_OSNet_new
在这里插入图片描述

测试

直接打开摄像头开始测试

测试bytetrack

python track.py --tracking-method bytetrack

测试结果如下:
在这里插入图片描述

测试strongsort

执行命令:

python track.py --tracking-method strongsort

在这里插入图片描述

测试ocsort

执行命令

python track.py --tracking-method ocsort

在这里插入图片描述

加入reid,并指定class

 python track.py --source 0 --reid-weights weights/osnet_x0_25_market1501.pt --classes 0

在这里插入图片描述

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。