尤物视频网站_国产精品成人在亚洲_国产成人亚洲综合无码不卡精品_丁香五月香婷婷五月_亚洲成AV人片高潮喷水

您的位置:首頁 >聚焦 >

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

2023-02-25 07:21:58    來源:程序員客棧

點擊下方卡片,關注「集智書童」公眾號


【資料圖】

點擊加入「集智書童-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加入到不同的跟蹤器之中里進行視頻流的檢測和跟蹤。

示例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視覺(分割、檢測)、多模態(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學習路上的「武功秘籍」

關鍵詞: 配置文件 多目標跟蹤 武功秘籍

相關閱讀