Initial commit

README.md

@@ -0,0 +1,12 @@
+---
+library_name: UniDepth
+tags:
+- monocular-metric-depth-estimation
+- pytorch_model_hub_mixin
+- model_hub_mixin
+repo_url: https://github.com/lpiccinelli-eth/UniDepth
+---
+
+This model has been pushed to the Hub using **UniDepth**:
+- Repo: https://github.com/lpiccinelli-eth/UniDepth
+- Docs: [More Information Needed]

config.json

@@ -1,31 +1,33 @@
 {
-  "generic": {
-    "seed": 13
-  },
-  "training": {},
   "data": {
     "image_shape": [
       462,
       616
     ]
   },
+  "generic": {
+    "seed": 13
+  },
   "model": {
+    "expansion": 4,
     "name": "UniDepthV1",
     "num_heads": 8,
-    "expansion": 4,
     "pixel_decoder": {
-      "hidden_dim": 512,
       "depths": [
         3,
         2,
         1
       ],
-      "dropout": 0.0
+      "dropout": 0.0,
+      "hidden_dim": 512
     },
     "pixel_encoder": {
-      "name": "dinov2_vitl14",
-      "pretrained": null,
-      "patch_size": 14,
+      "depths": [
+        5,
+        12,
+        18,
+        24
+      ],
       "embed_dim": 1024,
       "embed_dims": [
         1024,
@@ -53,12 +55,10 @@
         1024,
         1024
       ],
-      "depths": [
-        5,
-        12,
-        18,
-        24
-      ]
+      "name": "dinov2_vitl14",
+      "patch_size": 14,
+      "pretrained": null
     }
-  }
+  },
+  "training": {}
 }