Add Pixio encoder

configs/model/encoder/pixio.yaml

@@ -0,0 +1,10 @@
+# UniCeption encoder string used for selecting encoder class (python3 -m uniception.models.encoders.list)
+encoder_str: "pixio"
+# Name of the encoder
+name: "pixio"
+# Data normalization type
+data_norm_type: "pixio"
+# Flag to indicate whether to use gradient checkpointing for encoder
+gradient_checkpointing: True
+# Flag to indicate whether model class uses torch hub
+uses_torch_hub: False

configs/model/pixio.yaml

@@ -0,0 +1,18 @@
+defaults:
+  - default
+  - encoder: pixio
+  - info_sharing: aat_ifr_24_layers
+  - pred_head: dpt_pose_scale
+  - task: images_only
+
+# String for model factory
+model_str: "pixio"
+# Model config
+model_config:
+  name: "pixio"
+  encoder_config: ${model.encoder}
+  info_sharing_config: ${model.info_sharing}
+  pred_head_config: ${model.pred_head}
+  geometric_input_config: ${model.task}
+# Image Normalization Type
+data_norm_type: ${model.encoder.data_norm_type}

mapanything/models/__init__.py

@@ -198,6 +198,9 @@ def init_model_from_config(
         "class_name": "VGGTWrapper",
     },
     # Add other model classes here
+    "pixio": {
+        "class": MapAnything,
+    },
 }