Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -196,5 +196,7 @@ module = [
"torchvision",
"torchvision.transforms",
"firerequests",
"scipy",
"scipy.optimize",
]
ignore_missing_imports = true
2 changes: 1 addition & 1 deletion test/core/test_tracker_integration.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
from trackers.eval import evaluate_mot_sequences
from trackers.io.mot import _load_mot_file, _mot_frame_to_detections, _MOTOutput

_TRACKER_IDS = ["sort", "bytetrack", "ocsort"]
_TRACKER_IDS = ["sort", "bytetrack", "ocsort", "botsort"]
_METRICS = ["CLEAR", "HOTA", "Identity"]
_TEST_DATA_DIR = Path(__file__).resolve().parent.parent / "data"

Expand Down
6 changes: 6 additions & 0 deletions test/data/tracker_expected_dancetrack.json
Original file line number Diff line number Diff line change
Expand Up @@ -16,5 +16,11 @@
"MOTA": 98.187,
"IDF1": 74.367,
"IDSW": 631
},
"botsort": {
"HOTA": 79.999,
"MOTA": 99.511,
"IDF1": 76.389,
"IDSW": 614
}
}
6 changes: 6 additions & 0 deletions test/data/tracker_expected_sportsmot.json
Original file line number Diff line number Diff line change
Expand Up @@ -16,5 +16,11 @@
"MOTA": 97.791,
"IDF1": 79.21,
"IDSW": 917
},
"botsort": {
"HOTA": 85.544,
"MOTA": 98.925,
"IDF1": 80.53,
"IDSW": 1107
}
}
2 changes: 2 additions & 0 deletions trackers/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
from __future__ import annotations

from trackers.annotators.trace import MotionAwareTraceAnnotator
from trackers.core.botsort.tracker import BoTSORTTracker
from trackers.core.bytetrack.tracker import ByteTrackTracker
from trackers.core.ocsort.tracker import OCSORTTracker
from trackers.core.sort.tracker import SORTTracker
Expand All @@ -22,6 +23,7 @@
from trackers.utils.converters import xcycsr_to_xyxy, xyxy_to_xcycsr

__all__ = [
"BoTSORTTracker",
"ByteTrackTracker",
"CoordinatesTransformation",
"Dataset",
Expand Down
8 changes: 8 additions & 0 deletions trackers/core/botsort/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
# ------------------------------------------------------------------------
# Trackers
# Copyright (c) 2026 Roboflow. All Rights Reserved.
# Licensed under the Apache License, Version 2.0 [see LICENSE for details]
# ------------------------------------------------------------------------
from .tracker import BoTSORTTracker

__all__ = ["BoTSORTTracker"]
Loading