Add ONNX model inference implementations for MoGe-2, BEN2, and DA3

- Introduced `onnx_moge2.py` for MoGe-2 ONNX inference with NumPy post-processing.
- Added `onnx_ben2.py` for BEN2 ONNX inference with Pillow and NumPy preprocessing.
- Created `onnx_da3.py` for Depth Anything 3 inference with ONNX Runtime and NumPy.
- Implemented shared ONNX Runtime session helpers in `onnx_runtime.py`.
- Developed unit tests for server functionalities in `test_server.py`, including model caching and job management.
- Enhanced job server to handle frame extraction without requiring models.
This commit is contained in:
2026-08-17 10:37:55 +08:00
parent b3a07cde32
commit c5b77c5c85
58 changed files with 5840 additions and 9931 deletions

View File

@@ -15,28 +15,36 @@ uv run python pack.py
├── src/
│ └── snapform/
│ ├── blender_manifest.toml
│ ├── install_env.py
│ ├── models.py
│ ├── operators.py
│ ├── panel.py
│ ├── preferences.py
│ ├── properties.py
│ ├── client.py
│ ├── job_runtime.py
│ ├── blendjob/
│ │ ├── client.py
│ │ ├── controller.py
│ │ ├── environment.py
│ │ ├── installer/environment.py
│ │ ├── operator.py
│ │ ├── service.py
│ │ ├── runner.py
│ │ ├── runtime.py
│ │ └── server.py
│ ├── blender/
│ │ ├── depth_plane.py
│ │ └── surface.py
│ └── ai_service/
│ ├── server.py
│ └── server/
│ ├── __init__.py
│ ├── model_catalog.py
│ ├── model_download.py
│ ├── model_manager.py
│ ├── jobs.py
│ ├── ben2_onnx.py
│ ├── da3_onnx.py
│ ├── moge_onnx.py
│ ├── media.py
│ ├── geometry.py
│ ├── background.py
│ ├── moge2.py
│ ├── da3.py
│ ├── onnx_ben2.py
│ ├── onnx_da3.py
│ ├── onnx_moge2.py
│ └── onnx_runtime.py
└── tests/
```