Update model_type to "fmpose3d_humans" across configuration and model files for consistency in the FMPose3D framework.

Author: xiu-cs

fmpose3d/common/arguments.py

@@ -74,8 +74,8 @@ def init(self):
         self.parser.add_argument("--model_dir", type=str, default="")
         # Optional: load model class from a specific file path
         self.parser.add_argument("--model_path", type=str, default="")
-        # Model registry name (e.g. "fmpose3d"); used instead of --model_path
-        self.parser.add_argument("--model_type", type=str, default="fmpose3d")
+        # Model registry name (e.g. "fmpose3d_humans"); used instead of --model_path
+        self.parser.add_argument("--model_type", type=str, default="fmpose3d_humans")
         self.parser.add_argument("--model_weights_path", type=str, default="")
 
         self.parser.add_argument("--post_refine_reload", action="store_true")

fmpose3d/common/config.py

@@ -19,14 +19,14 @@
 @dataclass
 class ModelConfig:
     """Model architecture configuration."""
-    model_type: str = "fmpose3d"
+    model_type: str = "fmpose3d_humans"
 
 
 # Per-model-type defaults for fields marked with INFER_FROM_MODEL_TYPE.
 # Also consumed by PipelineConfig.for_model_type to set cross-config
 # values (dataset, sample_steps, etc.).
 _FMPOSE3D_DEFAULTS: Dict[str, Dict] = {
-    "fmpose3d": {
+    "fmpose3d_humans": {
         "n_joints": 17,
         "out_joints": 17,
         "dataset": "h36m",
@@ -39,7 +39,7 @@ class ModelConfig:
         "n_joints": 26,
         "out_joints": 26,
         "dataset": "animal3d",
-        "sample_steps": 3,
+        "sample_steps": 5,
         "joints_left": [0, 3, 5, 8, 10, 12, 14, 16, 20, 22],
         "joints_right": [1, 4, 6, 9, 11, 13, 15, 17, 21, 23],
         "root_joint": 7,
@@ -51,7 +51,7 @@ class ModelConfig:
 
 @dataclass
 class FMPose3DConfig(ModelConfig):
-    model_type: str = "fmpose3d"
+    model_type: str = "fmpose3d_humans"
     model: str = ""
     layers: int = 5
     channel: int = 512
@@ -321,7 +321,7 @@ def _pick(dc_class, src: dict):
 
         kwargs = {}
         for group_name, dc_class in _SUB_CONFIG_CLASSES.items():
-            if group_name == "model_cfg" and raw.get("model_type", 'fmpose3d') in _FMPOSE3D_DEFAULTS:
+            if group_name == "model_cfg" and raw.get("model_type", 'fmpose3d_humans') in _FMPOSE3D_DEFAULTS:
                 dc_class = FMPose3DConfig
             elif group_name == "pose2d_cfg":
                 p2d = raw.get("pose2d_model", "hrnet")

fmpose3d/models/fmpose3d/model_GAMLP.py

@@ -212,7 +212,7 @@ def forward(self, x):
         x = self.fc2(x)
         return x
 
-@register_model("fmpose3d")
+@register_model("fmpose3d_humans")
 class Model(BaseModel):
     def __init__(self, args):
         super().__init__(args)

tests/test_config.py

@@ -250,7 +250,7 @@ def test_from_namespace_basic(self):
         ns = argparse.Namespace(
             # FMPose3DConfig
             model="test_model",
-            model_type="fmpose3d",
+            model_type="fmpose3d_humans",
             layers=5,
             channel=256,
             d_hid=512,