HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3
ultralytics-ascend/ultralytics/tracker
History
Glenn Jocher edd3ff1669
Update .pre-commit-config.yaml (#1026)
2023-02-17 22:26:40 +01:00
..
cfg Integrate ByteTracker and BoT-SORT trackers (#788) 2023-02-15 20:23:03 +04:00
trackers Update .pre-commit-config.yaml (#1026) 2023-02-17 22:26:40 +01:00
utils Update .pre-commit-config.yaml (#1026) 2023-02-17 22:26:40 +01:00
__init__.py ultralytics 8.0.40 TensorRT metadata and Results visualizer (#1014) 2023-02-17 20:06:06 +01:00
README.md Integrate ByteTracker and BoT-SORT trackers (#788) 2023-02-15 20:23:03 +04:00
track.py Update .pre-commit-config.yaml (#1026) 2023-02-17 22:26:40 +01:00

README.md

Tracker

Trackers

  • ByteTracker
  • BoT-SORT

Usage

python interface:

from ultralytics import YOLO

model = YOLO("yolov8n.pt")  # or a segmentation model .i.e yolov8n-seg.pt
model.track(
    source="video/streams",
    stream=True,
    tracker="botsort.yaml/bytetrack.yaml",
    ...,
)

cli:

yolo detect track source=... tracker=...
yolo segment track source=... tracker=...

By default, trackers will use the configuration in ultralytics/tracker/cfg. We also support using a modified tracker config file. Please refer to the tracker config files in ultralytics/tracker/cfg.