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": "convnext_large",
-      "pretrained": null,
-      "patch_size": 16,
+      "depths": [
+        3,
+        6,
+        33,
+        36
+      ],
       "embed_dim": 192,
       "embed_dims": [
         192,
@@ -65,12 +67,10 @@
         1536,
         1536
       ],
-      "depths": [
-        3,
-        6,
-        33,
-        36
-      ]
+      "name": "convnext_large",
+      "patch_size": 16,
+      "pretrained": null
     }
-  }
+  },
+  "training": {}
 }