ultralytics-ascend/ultralytics/tracker
Glenn Jocher 28e48be5b6
ultralytics 8.0.56 PyTorch 2.0 support and minor fixes (#1538)
Co-authored-by: N-Friederich <127681326+N-Friederich@users.noreply.github.com>
Co-authored-by: Uhrendoktor <36703334+Uhrendoktor@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Aman Agarwal <amanag.11@gmail.com>
Co-authored-by: ExtReMLapin <3909752+ExtReMLapin@users.noreply.github.com>
Co-authored-by: Nadav Eidelstein <30617226+nodeav@users.noreply.github.com>
2023-03-24 00:22:20 +01:00
..
cfg ultralytics 8.0.42 DDP fix and Docs updates (#1065) 2023-02-20 12:56:20 +01:00
trackers ultralytics 8.0.56 PyTorch 2.0 support and minor fixes (#1538) 2023-03-24 00:22:20 +01:00
utils ultralytics 8.0.52 reduced TAL CUDA usage and AMP check fix (#1333) 2023-03-10 03:27:06 +01:00
__init__.py ultralytics 8.0.47 Docker and reformat updates (#1153) 2023-02-25 22:49:19 -08:00
README.md ultralytics 8.0.42 DDP fix and Docs updates (#1065) 2023-02-20 12:56:20 +01:00
track.py ultralytics 8.0.49 task, exports and metadata updates (#1197) 2023-03-01 21:16:09 -08:00

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",  # or '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.