vnguyen/product-recognition-demo
1
0
Fork 1
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

fix: use the tracker imports that exist in ultralytics 8.4.153

Browse source 
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>
This commit is contained in:
kiyo9w 2026-09-24 13:36:50 +09:00
parent 30c0292ed1
commit 5f93f6e278
1 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
recognization/code/tracking.py
View file

@ -28,9 +28,8 @@ current_dir = os.path.dirname(os.path.abspath(__file__))
if current_dir not in sys.path: if current_dir not in sys.path:
sys.path.insert(0, current_dir) sys.path.insert(0, current_dir)
from ultralytics.trackers.basetrack import BaseTrack, STrack from ultralytics.trackers.byte_tracker import BYTETracker, STrack
from ultralytics.trackers.byte_tracker import BYTETracker from ultralytics.utils import YAML, IterableSimpleNamespace
from ultralytics.utils import IterableSimpleNamespace, yaml_load
_BYTETRACK_YAML = os.path.join( _BYTETRACK_YAML = os.path.join(
os.path.dirname(os.path.abspath(__import__("ultralytics").__file__)), os.path.dirname(os.path.abspath(__import__("ultralytics").__file__)),
@ -50,7 +49,7 @@ DEFAULT_TRACKER_CONFIG = {
def _default_args() -> IterableSimpleNamespace: def _default_args() -> IterableSimpleNamespace:
try: try:
return IterableSimpleNamespace(**yaml_load(_BYTETRACK_YAML)) return IterableSimpleNamespace(**YAML.load(_BYTETRACK_YAML))
except Exception: except Exception:
return IterableSimpleNamespace(**DEFAULT_TRACKER_CONFIG) return IterableSimpleNamespace(**DEFAULT_TRACKER_CONFIG)