icrdr/AnyImage
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
- 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
Download Patch File Download Diff File Expand all files Collapse all files

19
tests/test_pack.py
View File

@@ -58,8 +58,23 @@ class PackTest(unittest.TestCase):
"assets/O_SnapForm.blend",
names,
)
self.assertIn("ai_service/jobs.py", names)
self.assertIn("ai_service/server.py", names)
self.assertIn("server/media.py", names)
self.assertIn("server/geometry.py", names)
self.assertIn("server/background.py", names)
self.assertIn("server/moge2.py", names)
self.assertIn("server/da3.py", names)
self.assertNotIn("server/processing.py", names)
self.assertNotIn("server/progress.py", names)
self.assertIn("server/model_catalog.py", names)
self.assertIn("server/model_download.py", names)
self.assertIn("server/onnx_ben2.py", names)
self.assertIn("server/onnx_da3.py", names)
self.assertIn("server/onnx_moge2.py", names)
self.assertNotIn("server/jobs.py", names)
self.assertIn("server/__init__.py", names)
self.assertIn("blendjob/environment.py", names)
self.assertIn("blendjob/installer/environment.py", names)
self.assertNotIn("install_env.py", names)
self.assertNotIn("snapform/__init__.py", names)
self.assertNotIn("importer.py", names)
self.assertFalse(