600字范文,内容丰富有趣,生活中的好帮手!
600字范文 > Yolov8的多目标跟踪实现

Yolov8的多目标跟踪实现

时间:2024-04-11 10:47:13

相关推荐

Yolov8的多目标跟踪实现

Yolov8_tracking

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 /mikel-brostrom/yolov8_tracking.git

安装

cd yolov8_trackingpip install -r requirements.txt

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

下载yolov8和REID权重

yolov8在这里可找到

/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外观特征有区别,不能完全说明问题。

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。