rmcewen/logo_test
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use script directory as base path for portability

Browse Source
This commit is contained in:
Rick McEwen
2026-01-06 16:00:09 -05:00
parent c086e8bbf7
commit 569285f664
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
prepare_test_data.py
View File

@ -113,11 +113,14 @@ def get_or_create_logo_name(cursor: sqlite3.Cursor, name: str) -> int:
def main(): def main():
# Paths # Use script directory as base path for portability
dataset_dir = Path("/data/dev.python/logo_test/LogoDet-3K") base_dir = Path(__file__).parent.resolve()
reference_dir = Path("/data/dev.python/logo_test/reference_logos")
test_images_dir = Path("/data/dev.python/logo_test/test_images") # Paths relative to script location
db_path = Path("/data/dev.python/logo_test/test_data_mapping.db") 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 # Ensure output directories exist
reference_dir.mkdir(exist_ok=True) reference_dir.mkdir(exist_ok=True)