feat: Add surface generation and media handling

- Implemented surface generation functionality in `surface.py`, including geometry estimation and texture writing.
- Introduced media input handling in `input.py`, supporting image and video frame extraction.
- Added model management for geometry adapters in `models/__init__.py`.
- Created BEN2 and DA3 model inference logic in `ben2.py` and `da3.py`, respectively.
- Developed ONNX model inference utilities for BEN2 and DA3 in `onnx_ben2.py` and `onnx_da3.py`.
- Added support for MoGe-2 model inference in `moge2.py` with necessary ONNX utilities.
- Implemented output handling for depth and surface data in `depth.py` and `surface.py`.
- Created debug output utilities for DA3 and MoGe-2 predictions in `debug_da3.py` and `debug_moge2.py`.
- Enhanced ONNX runtime session management in `onnx_runtime.py`.
- Updated `__init__.py` files to include new modules and functions for better organization.

src/snapform/server/model_manager.py

@@ -27,7 +27,7 @@ class ModelSessionCache:
         return self._load_ben2(model_dir, requested_device, key)
 
     def _load_ben2(self, model_dir, requested_device, key):
-        from .onnx_ben2 import create_session
+        from .models.onnx_ben2 import create_session
 
         started = time.perf_counter()
         model = create_session(model_dir, requested_device)
@@ -47,7 +47,7 @@ class ModelSessionCache:
         return self._load_da3(model_dir, requested_device, key)
 
     def _load_da3(self, model_dir, requested_device, key):
-        from .onnx_da3 import create_session
+        from .models.onnx_da3 import create_session
 
         model = create_session(model_dir, requested_device)
         self.da3_model = model
@@ -55,7 +55,7 @@ class ModelSessionCache:
         return model
 
     def get_moge2(self, model_dir, requested_device):
-        from .onnx_moge2 import create_session
+        from .models.onnx_moge2 import create_session
 
         key = (str(Path(model_dir).resolve()), requested_device, "onnx")
         if self.da3_model is not None: