文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Yolov8的多目标跟踪实现

2023-09-07 08:02

关注

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_trackingpip 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
运行效果,yolov8略好于yolov5,Strongsort略好于deepsort。
试验视频, MOT16-13-raw.mp4,与原来的REID模型ckpt.t7比较,osnet_x0_25, osnet_x1_0均不及ckpt.t7。
注:以上试验视频中跟踪目标类型为car,而非person,ckpt.t7和osnet模型都以person为训练集,与car外观特征有区别,不能完全说明问题。

来源地址:https://blog.csdn.net/weixin_44238733/article/details/129024558

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯