Skip to content

Commit 952e75b

Browse files
committed
refactor(depth-estimation): update models.json with ONNX format and variant structure
1 parent 6cf85cd commit 952e75b

1 file changed

Lines changed: 35 additions & 32 deletions

File tree

‎skills/transformation/depth-estimation/models.json‎

Lines changed: 35 additions & 32 deletions
Original file line numberDiff line numberDiff line change
@@ -59,31 +59,34 @@
5959
}
6060
},
6161
"linux": {
62-
"repository": "depth-anything/Depth-Anything-V2-Small",
63-
"format": "pth",
62+
"repository": "onnx-community/depth-anything-v2-small",
63+
"format": "onnx",
6464
"variants": {
65-
"depth_anything_v2_vits": {
65+
"model": {
6666
"precision": "float32",
67-
"size_mb": 99.0,
68-
"description": "PyTorch ViT-S — CUDA/CPU"
67+
"size_mb": 98.0,
68+
"description": "ONNX — CUDA/TensorRT/CPU"
69+
},
70+
"model_quantized": {
71+
"precision": "int8",
72+
"size_mb": 25.0,
73+
"description": "ONNX INT8 quantized — smallest, fastest"
6974
}
7075
}
7176
},
7277
"win32": {
73-
"repository": "depth-anything/Depth-Anything-V2-Small",
74-
"format": "pth",
78+
"repository": "onnx-community/depth-anything-v2-small",
79+
"format": "onnx",
7580
"variants": {
76-
"depth_anything_v2_vits": {
81+
"model": {
7782
"precision": "float32",
78-
"size_mb": 99.0,
79-
"description": "PyTorch ViT-S — CUDA/CPU"
83+
"size_mb": 98.0,
84+
"description": "ONNX — CUDA/TensorRT/DirectML/CPU"
8085
},
81-
"depth_anything_v2_vits_trt_fp16": {
82-
"precision": "float16",
83-
"size_mb": 52.6,
84-
"format": "trt",
85-
"requires": "tensorrt",
86-
"description": "TensorRT FP16 — built locally, 6.9x faster"
86+
"model_quantized": {
87+
"precision": "int8",
88+
"size_mb": 25.0,
89+
"description": "ONNX INT8 quantized — smallest, fastest"
8790
}
8891
}
8992
}
@@ -96,24 +99,24 @@
9699
"input_size": [518, 392],
97100
"platforms": {
98101
"linux": {
99-
"repository": "depth-anything/Depth-Anything-V2-Base",
100-
"format": "pth",
102+
"repository": "onnx-community/depth-anything-v2-base",
103+
"format": "onnx",
101104
"variants": {
102-
"depth_anything_v2_vitb": {
105+
"model": {
103106
"precision": "float32",
104107
"size_mb": 390.0,
105-
"description": "PyTorch ViT-B — CUDA/CPU"
108+
"description": "ONNX — CUDA/TensorRT/CPU"
106109
}
107110
}
108111
},
109112
"win32": {
110-
"repository": "depth-anything/Depth-Anything-V2-Base",
111-
"format": "pth",
113+
"repository": "onnx-community/depth-anything-v2-base",
114+
"format": "onnx",
112115
"variants": {
113-
"depth_anything_v2_vitb": {
116+
"model": {
114117
"precision": "float32",
115118
"size_mb": 390.0,
116-
"description": "PyTorch ViT-B — CUDA/CPU"
119+
"description": "ONNX — CUDA/TensorRT/DirectML/CPU"
117120
}
118121
}
119122
}
@@ -126,24 +129,24 @@
126129
"input_size": [518, 392],
127130
"platforms": {
128131
"linux": {
129-
"repository": "depth-anything/Depth-Anything-V2-Large",
130-
"format": "pth",
132+
"repository": "onnx-community/depth-anything-v2-large",
133+
"format": "onnx",
131134
"variants": {
132-
"depth_anything_v2_vitl": {
135+
"model": {
133136
"precision": "float32",
134137
"size_mb": 1280.0,
135-
"description": "PyTorch ViT-L — CUDA recommended"
138+
"description": "ONNX — CUDA/TensorRT/CPU"
136139
}
137140
}
138141
},
139142
"win32": {
140-
"repository": "depth-anything/Depth-Anything-V2-Large",
141-
"format": "pth",
143+
"repository": "onnx-community/depth-anything-v2-large",
144+
"format": "onnx",
142145
"variants": {
143-
"depth_anything_v2_vitl": {
146+
"model": {
144147
"precision": "float32",
145148
"size_mb": 1280.0,
146-
"description": "PyTorch ViT-L — CUDA recommended"
149+
"description": "ONNX — CUDA/TensorRT/DirectML/CPU"
147150
}
148151
}
149152
}

0 commit comments

Comments
 (0)