当前位置: 首页 > news >正文

Yolov8的多目标跟踪实现

Yolov8_tracking
2023年2月,Yolov5发展到yolov8,这世界变得真快哦。Yolov8由ultralytics公司发布,yolov6-美团,yolov7-Alexey Bochkovskiy和Chien-Yao Wang,其各有高招,对yolov5均有提升。mikel-brostrom在github上不断更新多目标跟踪方法,deepsort升级到StrongSort,检测用yolov8,tracker除了StrongSort外,还有 ocsort和bytetrack,眼花缭乱。
来体验一下mikel-brostrom提供的yolov8_tracking。
克隆yolov8_tracking

git clone --recurse-submodules https://github.com/mikel-brostrom/yolov8_tracking.git

安装

cd yolov8_tracking
pip install -r requirements.txt  

安装python和pytorch时要解决各包间的版本依赖关系,这往往要花费时间,因为不同的硬件平台,linux版本,python版本,python调试工具各不相同。

下载yolov8和REID权重
yolov8在这里可找到
https://github.com/ultralytics/ultralytics/blob/main/README.zh-CN.md

下载,将yolov8n.pt和yolov8s.pt放到目录 yolov8_tracking/yolov8/weights。Reid特征识别用osnet,在安装目录yolov8_tracking/trackers/strongsort/deep/checkpoint下已有如下REID权重:
osnet_x0_25_market1501.pth
osnet_x0_25_msmt17.pth
osnet_x1_0_msmt17.pth
yolov8_tracking只支持后缀为pt的REID权重文件,这里选择osnet_x0_25_market1501.pth,修改后缀为pt。
命令行

track.py --yolo-weight yolov8/yolov8n.pt \--tracking-method strongsort    \--source MOT16-13-raw.mp4   \--img 640     \--reid-weights trackers/strongsort/deep/checkpoint/osnet_x0_25_market1501.pt   \--classes 0 2   \

试验过程没有遇到问题。本人用lenovo thinkpad E560,没有nvidia显卡支持,只能用device=cpu方式运行跟踪推理过程,运行速度比较慢,实际运行结果如下:
Yolov8n, osnet_x0_25_market1501, 640x384视频,每帧处理时间0.2s
Yolov8s, osnet_x0_25_market1501, 1280x736视频, 每帧处理时间1.2s
Yolov8n, osnet_x0_25_market1501, 1280x736视频,每帧处理时间0.5s
运行效果,明显比以前的deepsort好,即使用yolov8n,也比deepsort yolov5m强,说明strongsort比deepsort,yolov8比yolov5都有效果的提升。

http://www.lryc.cn/news/5967.html

相关文章:

  • 28--Django-后端开发-drf之自定义全局异常、接口文档生成以及三大认证源码分析
  • 【MyBatis】动态SQL
  • LeetCode(剑指offer) Day1
  • 1、MyBatis框架——JDBC代码回顾与分析、lombok插件的安装与使用
  • 笔记-GPS设备定位方式
  • 2023秋招携程SRE算法岗面试经验分享
  • 4.9 内部类
  • ncnn模型精度验证
  • IB-PYP幼儿十大素质培养目标
  • 02.13:监督学习中的分类问题
  • leetcode刷题 | 关于二叉树的题型总结3
  • 设计模式-结构型
  • 【新】华为OD机试 - 预订酒店(Python)| 运气好 会考到原题
  • 【编程基础之Python】4、安装Python开发工具
  • 5. 最长回文子串
  • 内网渗透(二十四)之Windows协议认证和密码抓取-Mimikatz读取sam和lsass获取密码
  • 【THREE.JS】网页中的炫酷3D
  • Go语言之 下载安装go以及vscode配置go环境
  • RBAC权限 API声明四种kubernetes对象
  • CDGP仿真选择题4
  • 典型相关分析与R语言实现
  • 【蓝桥集训】第一天——前缀和
  • 2022-03-19青少年软件编程(C语言)等级考试试卷(六级)解析
  • [JavaScript 刷题] 特殊数组的特征值, leetcode 1608
  • 各种素材网站大全【全部倾倒,福利倒计时-JS,HTML,游戏素材,UI,图片素材等
  • 影片自由,丝滑流畅,Docker容器基于WebDav协议通过Alist挂载(百度网盘/阿里云盘)Python3.10接入
  • 【新】华为OD机试 - 数组的中心位置(Python)| 运气好,这就是原题
  • 小米电视安装 Plex 打造家庭影院
  • Elasticsearch:Combined fields 查询
  • uart 子系统