Add generated VISLOC examples for Space

.gitignore

@@ -2,5 +2,6 @@ __pycache__/
 .DS_Store
 .python-version
 data/
+examples/
 *.pyc
 scripts/

README.md

@@ -15,6 +15,7 @@ Analyze aerial RGB imagery to detect safe drone landing sites. Combines monocula
 
 ## What’s inside
 - **Main app (`app.py`)** — runs full inference with adjustable thresholds, overlays, and camera assumptions; requires >8GB VRAM (assuming default 1024 px processing resolution); runtime is ~2000 ms per image.
+- **Sample data (`examples/visloc/`)** — 5 VISLOC frames bundled for quick testing in the Space.
 
 ## References
 - UAV-VisLoc dataset: Xu et al., 2024 (https://arxiv.org/abs/2405.11936)

app/config.py

@@ -5,6 +5,7 @@ from pathlib import Path
 
 VISLOC_DIR = Path("data/Image/VISLOC")
 IMAGE_ROOT = Path("data/Image")
+EXAMPLES_ROOT = Path("examples")
 IMAGE_EXTS = (".jpg", ".jpeg", ".png", ".JPG", ".JPEG", ".PNG")
 DEFAULT_ALTITUDE_M = 450.0
 ASSUMED_FOV_DEG = 90.0
@@ -51,6 +52,7 @@ DEFAULT_ANALYZER_SETTINGS = AnalyzerSettings()
 __all__ = [
     "VISLOC_DIR",
     "IMAGE_ROOT",
+    "EXAMPLES_ROOT",
     "IMAGE_EXTS",
     "DEFAULT_ALTITUDE_M",
     "ASSUMED_FOV_DEG",

app/data_sources.py

@@ -3,7 +3,8 @@ from __future__ import annotations
 from functools import lru_cache
 from pathlib import Path
 
-from .config import IMAGE_EXTS, IMAGE_ROOT, VISLOC_DIR
+from .config import EXAMPLES_ROOT, IMAGE_EXTS, IMAGE_ROOT, VISLOC_DIR
+from .example_assets import ensure_example_images
 
 
 @lru_cache(maxsize=1)
@@ -14,6 +15,16 @@ def list_visloc_images() -> list[Path]:
     return sorted(files)
 
 
+@lru_cache(maxsize=1)
+def list_example_images() -> list[Path]:
+    examples_dir = EXAMPLES_ROOT / "visloc"
+    ensure_example_images(examples_dir)
+    if not examples_dir.exists():
+        return []
+    files = [p for p in examples_dir.rglob("*") if p.is_file() and p.suffix in IMAGE_EXTS]
+    return sorted(files)
+
+
 @lru_cache(maxsize=1)
 def list_all_data_inputs() -> list[str]:
     paths: list[Path] = []
@@ -24,6 +35,7 @@ def list_all_data_inputs() -> list[str]:
                 paths.append(p)
 
     # Prefer structured datasets first
+    _add(list_example_images())
     _add(list_visloc_images())
 
     # Allow arbitrary images anywhere under data/Image/
@@ -38,11 +50,13 @@ def list_all_data_inputs() -> list[str]:
 
 def clear_caches() -> None:
     list_visloc_images.cache_clear()
+    list_example_images.cache_clear()
     list_all_data_inputs.cache_clear()
 
 
 __all__ = [
     "list_visloc_images",
+    "list_example_images",
     "list_all_data_inputs",
     "clear_caches",
 ]