Update model/run_inference.py

model/run_inference.py

@@ -29,6 +29,9 @@ from mmengine.registry import init_default_scope
 from mmdet3d.utils import register_all_modules
 register_all_modules(init_default_scope=False)
 
+import importlib, sys
+from pathlib import Path
+
 @lru_cache(maxsize=2)
 def get_model(model_key: str):
     if model_key not in MODEL_REGISTRY:
@@ -40,6 +43,23 @@ def get_model(model_key: str):
     else:  # detr3d
         purge_project_registrations("projects.PETR")
 
+    # Ensure the mmdetection3d repo root (which contains `projects/`) is importable
+    repo_root = Path(__file__).resolve().parents[1] / "mmdetection3d"
+    if repo_root.is_dir() and str(repo_root) not in sys.path:
+        sys.path.insert(0, str(repo_root))
+    
+    # Import the correct project so its registries (e.g., ResizeCropFlipImage) are registered
+    proj_name = "projects.PETR.petr" if "petr" in model_key.lower() else "projects.DETR3D.detr3d"
+    try:
+        importlib.import_module(proj_name)
+    except ModuleNotFoundError as e:
+        # Helpful error that tells you what path we tried
+        raise ModuleNotFoundError(
+            f"Could not import {proj_name}. Ensure the 'projects' package is on sys.path. "
+            f"Tried adding: {repo_root}"
+        ) from e
+    
+
     config_path, (repo_id, hf_file) = MODEL_REGISTRY[model_key]
     if not os.path.isfile(config_path):
         raise FileNotFoundError(f"Config not found: {config_path}")