From 5f93f6e2781da9b8f420d36260182c13aaaa50cc Mon Sep 17 00:00:00 2001 From: kiyo9w Date: Thu, 24 Sep 2026 13:36:50 +0900 Subject: [PATCH] fix: use the tracker imports that exist in ultralytics 8.4.153 STrack lives in ultralytics.trackers.byte_tracker and the YAML loader is ultralytics.utils.YAML; the basetrack/yaml_load names this module used are gone in the pinned version. Caught by running the tracker tests against the real dependency. Co-Authored-By: Devin <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- recognization/code/tracking.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/recognization/code/tracking.py b/recognization/code/tracking.py index 01699ba..a77bc61 100644 --- a/recognization/code/tracking.py +++ b/recognization/code/tracking.py @@ -28,9 +28,8 @@ current_dir = os.path.dirname(os.path.abspath(__file__)) if current_dir not in sys.path: sys.path.insert(0, current_dir) -from ultralytics.trackers.basetrack import BaseTrack, STrack -from ultralytics.trackers.byte_tracker import BYTETracker -from ultralytics.utils import IterableSimpleNamespace, yaml_load +from ultralytics.trackers.byte_tracker import BYTETracker, STrack +from ultralytics.utils import YAML, IterableSimpleNamespace _BYTETRACK_YAML = os.path.join( os.path.dirname(os.path.abspath(__import__("ultralytics").__file__)), @@ -50,7 +49,7 @@ DEFAULT_TRACKER_CONFIG = { def _default_args() -> IterableSimpleNamespace: try: - return IterableSimpleNamespace(**yaml_load(_BYTETRACK_YAML)) + return IterableSimpleNamespace(**YAML.load(_BYTETRACK_YAML)) except Exception: return IterableSimpleNamespace(**DEFAULT_TRACKER_CONFIG)