YOLOv8官方支持多目标跟踪 | ByteTrack、BoT-SORT都已加入YOLOv8官方

2023-02-25 23:37:42 [来源:程序员客栈]

点击下方卡片,关注「集智书童」公众号

点击加入「集智书童-YOLO算法」交流群

模板跟踪是一项任务,涉及识别模板的位置和类别,然后为视频流中的检测分配唯一ID。跟踪器的输出与添加了模板ID的检测相同。


(资料图片仅供参考)

YOLOv8加入了哪些检测器?

以下跟踪算法已经实现,可以通过 tracker=tracker_type.yaml实现:

BoT-SORT - botsort.yaml

ByteTrack - bytetrack.yaml

默认跟踪器为:BoT-SORT

Tracking

将训练好的 YOLOv8n/YOLOv8n-seg model加入到不同的跟踪器之中里进行视频流的检测和跟踪。

示例1

fromultralyticsimportYOLO#Loadamodelmodel=YOLO("yolov8n.pt")#loadanofficialdetectionmodelmodel=YOLO("yolov8n-seg.pt")#loadanofficialsegmentationmodelmodel=YOLO("path/to/best.pt")#loadacustommodel#Trackwiththemodelresults=model.track(source="https://youtu.be/Zgi9g1ksQHc",show=True)results=model.track(source="https://youtu.be/Zgi9g1ksQHc",show=True,tracker="bytetrack.yaml")

命令行如下

yolotrackmodel=yolov8n.ptsource="https://youtu.be/Zgi9g1ksQHc"#officialdetectionmodelyolotrackmodel=yolov8n-seg.ptsource=...#officialsegmentationmodelyolotrackmodel=path/to/best.ptsource=...#custommodelyolotrackmodel=path/to/best.pttracker="bytetrack.yaml"#bytetracktracker

与上述用法一样,YOLOv8支持用于跟踪的检测和分割模型,只需加载相应的(检测或分割)模型即可。

配置跟踪

跟踪与预测共享配置,即“conf”、“iou”、“show”。更多配置请参考 predict page。

示例1

fromultralyticsimportYOLOmodel=YOLO("yolov8n.pt")results=model.track(source="https://youtu.be/Zgi9g1ksQHc",conf=0.3,iou=0.5,show=True)

命令行如下

yolotrackmodel=yolov8n.ptsource="https://youtu.be/Zgi9g1ksQHc"conf=0.3,iou=0.5show

跟踪器

YOLOv8还支持使用修改的跟踪器配置文件,只需复制一个配置文件即可,比如复制 custom_tracker.yamlultralytics/tracker/cfg并修改配置(比如 tracker_type)。

示例2

fromultralyticsimportYOLOmodel=YOLO("yolov8n.pt")results=model.track(source="https://youtu.be/Zgi9g1ksQHc",tracker="custom_tracker.yaml")

命令行如下

yolotrackmodel=yolov8n.ptsource="https://youtu.be/Zgi9g1ksQHc"tracker="custom_tracker.yaml"

具体可以参考ultralytics/tracker/cfg。

参考

[1].https://github.com/ultralytics/ultralytics.

推荐阅读

YOLOv5抛弃Anchor-Base方法 | YOLOv5u正式加入Anchor-Free大家庭

改进YOLOX | Push-IOU+Dynamic Anchor进一步提升YOLOX性能

即插即用 | CFNet提出全新多尺度融合方法!显著提升检测和分割精度!

扫码加入「集智书童-YOLO算法」交流群

(备注:方向+学校/公司+昵称)

想要了解更多:

前沿AI视觉感知全栈知识「分类、检测、分割、关键点、车道线检测、3D视觉(分割、检测)、多模态、目标跟踪、NerF」

行业技术方案「AI安防、AI医疗、AI自动驾驶」

AI模型部署落地实战「CUDA、TensorRT、NCNN、OpenVINO、MNN、ONNXRuntime以及地平线框架」

欢迎扫描上方二维码,加入「集智书童-知识星球」,日常分享论文、学习笔记、问题解决方案、部署方案以及全栈式答疑,期待交流!

免责声明

凡本公众号注明“来源:XXX(非集智书童)”的作品,均转载自其它媒体,版权归原作者所有,如有侵权请联系我们删除,谢谢。

点击下方“阅读原文”,

了解更多AI学习路上的「武功秘籍」

x 广告
x 广告
© 2021 南极早报网 版权所有,备案号:粤ICP备2022077823号-13邮箱: 317 493 128@qq.com