YOLOv8官方支持多目標跟蹤 | ByteTrack、BoT-SORT都已加入YOLOv8官方
點擊下方卡片,關注「集智書童」公眾號
【資料圖】
點擊加入「集智書童-YOLO算法」交流群
模板跟蹤是一項任務,涉及識別模板的位置和類別,然后為視頻流中的檢測分配唯一ID。跟蹤器的輸出與添加了模板ID的檢測相同。
YOLOv8加入了哪些檢測器?以下跟蹤算法已經(jīng)實現(xiàn),可以通過 tracker=tracker_type.yaml實現(xiàn):
BoT-SORT - botsort.yamlByteTrack - bytetrack.yaml默認跟蹤器為:BoT-SORT
Tracking將訓練好的 YOLOv8n/YOLOv8n-seg model加入到不同的跟蹤器之中里進行視頻流的檢測和跟蹤。
示例1fromultralyticsimportYOLO#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。
示例1fromultralyticsimportYOLOmodel=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)。
示例2fromultralyticsimportYOLOmodel=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視覺(分割、檢測)、多模態(tài)、目標跟蹤、NerF」
行業(yè)技術(shù)方案「AI安防、AI醫(yī)療、AI自動駕駛」AI模型部署落地實戰(zhàn)「CUDA、TensorRT、NCNN、OpenVINO、MNN、ONNXRuntime以及地平線框架」歡迎掃描上方二維碼,加入「集智書童-知識星球」,日常分享論文、學習筆記、問題解決方案、部署方案以及全棧式答疑,期待交流!
免責聲明凡本公眾號注明“來源:XXX(非集智書童)”的作品,均轉(zhuǎn)載自其它媒體,版權(quán)歸原作者所有,如有侵權(quán)請聯(lián)系我們刪除,謝謝。點擊下方“閱讀原文”,了解更多AI學習路上的「武功秘籍」相關閱讀
-
YOLOv8官方支持多目標跟蹤 | ByteTrac...
點擊下方卡片,關注「集智書童」公眾號點擊加入「集智書童-YOLO算法... -
老胡已瘋,他說公務員加班太辛苦|全球球...
拿什么來形容一下胡錫進呢?不太好說,他似乎陷入了某種混沌不清的... -
推薦十個優(yōu)秀的富文本編輯器 今日看點
前言富文本編輯器是一種可嵌入瀏覽器網(wǎng)頁中,所見即所得的文本編輯... -
世界新資訊:你的用戶珍貴么?殺雞吃肉...
最近越來越感受到,“用戶是否珍貴”,是考慮產(chǎn)品的各種策略時一個... -
建議收藏,輕松搞懂區(qū)塊鏈! 全球熱文
未來已來,只是不均衡地分布在當下大家好,我是菜農(nóng),歡迎來到我的... -
酷睿i5-13490F評測體驗:延續(xù)小黑盒,性...
在全新架構(gòu)和制程工藝的加持下,去年12代酷睿處理器的表現(xiàn)非常的亮...