CodeChefJeff commited on
Commit
99b7fe5
·
verified ·
1 Parent(s): 5e82fc3

Prebuilt TensorRT 10.7 engine for Jetson Orin (sm_87)

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ da2_metric_hypersim_vits_364x448_trt100700_sm87.engine filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -15,5 +15,8 @@ Depth Anything V2 Small, metric, Hypersim (indoor, max depth 20 m), converted fr
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
 
19
  The crate pins these files by sha256 in its `model.json`. Weights are Apache-2.0, as the upstream Small model.
 
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.
21
 
22
  The crate pins these files by sha256 in its `model.json`. Weights are Apache-2.0, as the upstream Small model.
da2_metric_hypersim_vits_364x448_trt100700_sm87.engine ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:64fab3e772a7d7cb8d5728d84b2e497db157065b97db5a33e72701a07e4ef6fe
3
+ size 53823004