您当前的位: 首页 > 宏观 >

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

来源:程序员客栈 2023-02-28 07:46:00

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

点击加入「集智书童-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学习路上的「武功秘籍」

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

点击下方卡片,关注「集智书童」公众号点击加入「集智书童-YOLO算法」交流群模板跟踪是一项任务,涉及识别模板的位置和类别,然后为视频流中的

程序员客栈 2023-02-28
94年属什么生肖婚配_94年属什么生肖适合的婚配_环球快看点

1、94年属狗,要配:虎、兔、好马,天作之合,处处成功,永远幸福,家财万贯。2、狗兔相合,最好找一只兔子,这是最好的婚姻

互联网 2023-02-28
当前热议!主耶稣我爱你歌词歌谱_主耶稣我爱你歌谱

1、主耶稣我爱你爱你,献上我真诚一片,如那满山的金达莱花,馨香把我环绕。2、我爱你我的心爱你,愿把终身献给你,愿我主紧紧

互联网 2023-02-27
螨性皮炎怎么治疗_螨虫性皮炎怎么办

螨虫性皮炎怎么办1、螨虫性皮炎怎么办首先要脱离有螨虫的环境,及时更衣洗澡。局部涂搽消炎止痒药,如1%酚或薄荷炉甘石洗剂,

互联网 2023-02-27
全球观点:当日快讯:香港特区政府欢迎日本与韩国当局撤销对香港客运航班的限制

据香港特区政府新闻公报,日本当局由去年12月底开始限制由香港出发的载客航班的增幅,香港特别行政区政府(特区政府)得悉日本当局3月1日起撤

互联网 2023-02-27
明星参加米兰时装周只是为了“时尚”吗?_全球头条

目前,米兰时装周正在火热进行中,来自世界各地的人汇聚于此,只为追求时尚品味的提升。明星作为时尚界的“杠把子”,自然是少不了要参加这种

穿搭小达人 2023-02-27

Copyright ©  2015-2022 太平洋数据网版权所有  备案号:豫ICP备2022016495号-17   联系邮箱:93 96 74 66 9@qq.com