commit success

TaskerJang · Jun 17, 2024 · fb4a17a · fb4a17a
1 parent 1412011
commit fb4a17a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/tracking/__init__.py b/tracking/__init__.py
diff --git a/tracking/our_track.py b/tracking/our_track.py
@@ -1,8 +1,15 @@
 import sys
 from pathlib import Path
 
-# 'tracking' 모듈 경로를 시스템 경로에 추가합니다.
-tracking_path = Path("K:/lost-and-find/tracking")
+# 현재 스크립트 파일의 위치를 기준으로 'tracking' 모듈 경로를 시스템 경로에 추가합니다.
+current_dir = Path(__file__).resolve().parent
+tracking_path = current_dir.parent  # 상위 디렉토리를 추가
+
+# 경로 출력 (디버깅용)
+print(f"Current directory: {current_dir}")
+print(f"Tracking path: {tracking_path}")
+
+# 시스템 경로에 추가
 sys.path.append(str(tracking_path))
 
 from tracking.detectors import get_yolo_inferer
@@ -27,7 +34,6 @@
 __tr = TestRequirements()
 __tr.check_packages(('ultralytics @ git+https://github.com/mikel-brostrom/ultralytics.git', ))  # install
 
-
 # BLIP 모델과 프로세서 초기화
 processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")
 model = BlipForConditionalGeneration.from_pretrained("Salesforce/blip-image-captioning-base")