点击下方卡片,关注「集智书童」公众号
点击加入「集智书童-YOLO算法」交流群
模板跟踪是一项任务,涉及识别模板的位置和类别,然后为视频流中的检测分配唯一ID。跟踪器的输出与添加了模板ID的检测相同。
(资料图片仅供参考)
YOLOv8加入了哪些检测器?以下跟踪算法已经实现,可以通过 tracker=tracker_type.yaml实现:
BoT-SORT - botsort.yaml
ByteTrack - 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视觉(分割、检测)、多模态、目标跟踪、NerF」
行业技术方案「AI安防、AI医疗、AI自动驾驶」
AI模型部署落地实战「CUDA、TensorRT、NCNN、OpenVINO、MNN、ONNXRuntime以及地平线框架」
欢迎扫描上方二维码,加入「集智书童-知识星球」,日常分享论文、学习笔记、问题解决方案、部署方案以及全栈式答疑,期待交流!
免责声明
凡本公众号注明“来源:XXX(非集智书童)”的作品,均转载自其它媒体,版权归原作者所有,如有侵权请联系我们删除,谢谢。
点击下方“阅读原文”,
了解更多AI学习路上的「武功秘籍」