ONNX export with a dynamic input size
Browse files- README.md +2 -1
- da2_metric_hypersim_vits.onnx +3 -0
README.md
CHANGED
|
@@ -14,7 +14,8 @@ Depth Anything V2 Small, metric, Hypersim (indoor, max depth 20 m), converted fr
|
|
| 14 |
|
| 15 |
- `dinov2_vits14.safetensors`, `da2_head_vits.safetensors`: the encoder and the DPT head, for candle
|
| 16 |
(converted by `tools/convert_weights.py`; includes the unused ImageNet classifier head candle's dinov2 loads).
|
| 17 |
-
- `
|
|
|
|
| 18 |
- `da2_metric_hypersim_vits_364x448_trt100700_sm87.engine`: that ONNX as an fp16 TensorRT 10.7 engine, built on a
|
| 19 |
Jetson AGX Orin (sm_87, JetPack 6 / L4T R36.3), so Orins skip the minutes-long engine build. Other TensorRT
|
| 20 |
versions or GPUs reject it and the crate builds its own from the ONNX.
|
|
|
|
| 14 |
|
| 15 |
- `dinov2_vits14.safetensors`, `da2_head_vits.safetensors`: the encoder and the DPT head, for candle
|
| 16 |
(converted by `tools/convert_weights.py`; includes the unused ImageNet classifier head candle's dinov2 loads).
|
| 17 |
+
- `da2_metric_hypersim_vits.onnx`: an ONNX export taking any input size that is a multiple of 14, for TensorRT
|
| 18 |
+
(made by the crate's `tools/export_onnx.py`). `da2_metric_hypersim_vits_364x448.onnx` is the older fixed-size one.
|
| 19 |
- `da2_metric_hypersim_vits_364x448_trt100700_sm87.engine`: that ONNX as an fp16 TensorRT 10.7 engine, built on a
|
| 20 |
Jetson AGX Orin (sm_87, JetPack 6 / L4T R36.3), so Orins skip the minutes-long engine build. Other TensorRT
|
| 21 |
versions or GPUs reject it and the crate builds its own from the ONNX.
|
da2_metric_hypersim_vits.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:beaeaeacdb66fc067fc82dd30463b255549367dc251d6cdff3db31bc3edf0750
|
| 3 |
+
size 101177651
|