From 569285f664b43f4b419613b02ee6cfc4150422bf Mon Sep 17 00:00:00 2001 From: Rick McEwen Date: Tue, 6 Jan 2026 16:00:09 -0500 Subject: [PATCH] Use script directory as base path for portability --- prepare_test_data.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/prepare_test_data.py b/prepare_test_data.py index ba4a524..7f96106 100755 --- a/prepare_test_data.py +++ b/prepare_test_data.py @@ -113,11 +113,14 @@ def get_or_create_logo_name(cursor: sqlite3.Cursor, name: str) -> int: def main(): - # Paths - dataset_dir = Path("/data/dev.python/logo_test/LogoDet-3K") - reference_dir = Path("/data/dev.python/logo_test/reference_logos") - test_images_dir = Path("/data/dev.python/logo_test/test_images") - db_path = Path("/data/dev.python/logo_test/test_data_mapping.db") + # Use script directory as base path for portability + base_dir = Path(__file__).parent.resolve() + + # Paths relative to script location + dataset_dir = base_dir / "LogoDet-3K" + reference_dir = base_dir / "reference_logos" + test_images_dir = base_dir / "test_images" + db_path = base_dir / "test_data_mapping.db" # Ensure output directories exist reference_dir.mkdir(exist_ok=True)