feat: Implement ONNX model inference for BEN2, Depth Anything 3, and MoGe
- Added `ben2_onnx.py` for BEN2 ONNX inference with preprocessing. - Introduced `da3_onnx.py` for Depth Anything 3 inference, including camera projection calculations. - Created `moge_onnx.py` for MoGe-2 ONNX inference and post-processing. - Implemented shared ONNX Runtime session helpers in `onnx_runtime.py`. - Added model synchronization script `sync_models.py` for downloading and verifying ONNX models. - Developed unit tests for BEN2, Depth Anything 3, and MoGe ONNX models to ensure functionality. - Updated `vendor.lock.toml` to remove unused artifacts.
This commit is contained in:
@@ -1,32 +1,5 @@
|
||||
schema_version = 1
|
||||
|
||||
[artifacts.depth-anything-3]
|
||||
revision = "3d835ec1a5802d64a8b8b15f817a1ab54809bfe4"
|
||||
filename = "depth-anything-3.tar.gz"
|
||||
url = "https://codeload.github.com/ByteDance-Seed/Depth-Anything-3/tar.gz/3d835ec1a5802d64a8b8b15f817a1ab54809bfe4"
|
||||
sha256 = "98aa2dd53ab44b96cef5190ae4841c6ae51797d099b784e08e12a1a55bd3a69b"
|
||||
patches = "depth-anything-3"
|
||||
|
||||
[artifacts.ben2]
|
||||
revision = "2c99a5da477b5523585bfa5c893888a6e818a8f6"
|
||||
filename = "ben2.tar.gz"
|
||||
url = "https://codeload.github.com/PramaLLC/BEN2/tar.gz/2c99a5da477b5523585bfa5c893888a6e818a8f6"
|
||||
sha256 = "01fd26716f3ff3a3f18ad2336c768af5b759edfc7c38bb3bf733b7c86eee6282"
|
||||
|
||||
[artifacts.moge]
|
||||
revision = "925b8ed835a7a9cdb7578ba15c658a0afc969030"
|
||||
filename = "moge.tar.gz"
|
||||
url = "https://codeload.github.com/microsoft/MoGe/tar.gz/925b8ed835a7a9cdb7578ba15c658a0afc969030"
|
||||
sha256 = "c6d90b2564d643f0f3ae65b5ffe163abaea3b55d194c6e2ba9976d67f552bd26"
|
||||
patches = "moge"
|
||||
|
||||
[artifacts.utils3d]
|
||||
revision = "3fab839f0be9931dac7c8488eb0e1600c236e183"
|
||||
filename = "utils3d.tar.gz"
|
||||
url = "https://codeload.github.com/EasternJournalist/utils3d/tar.gz/3fab839f0be9931dac7c8488eb0e1600c236e183"
|
||||
sha256 = "f1f167b1f68809bea391a66dd0d2ad0e14f7382cbddf087b655181c31a285abb"
|
||||
patches = "utils3d"
|
||||
|
||||
[uv]
|
||||
version = "0.11.3"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user