CodeChefJeff commited on
Commit
999a86e
·
verified ·
1 Parent(s): 99b7fe5

ONNX export with a dynamic input size

Browse files
Files changed (2) hide show
  1. README.md +2 -1
  2. 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
- - `da2_metric_hypersim_vits_364x448.onnx`: an ONNX export with a fixed 364x448 input, for TensorRT on Jetsons.
 
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