当前位置：首页 > 关注 > 正文

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

来源：程序员客栈    时间：2023-02-28 23:46:30

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


(资料图)

点击加入「集智书童-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 关闭

资讯

到2024年！昆明外贸进出口规模将突破3000亿元

行业

X 关闭

产业