[Fix] fix some bugs in pare configs (#188)

Co-authored-by: Feng <[email protected]>
open-mmlab · May 31, 2022 · ff99666 · ff99666
1 parent 499083c
commit ff99666
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/configs/pare/README.md b/configs/pare/README.md
@@ -49,7 +49,7 @@ mmhuman3d
     │       ├── SMPL_FEMALE.pkl
     │       ├── SMPL_MALE.pkl
     │       └── SMPL_NEUTRAL.pkl
-    ├── pretrained
+    ├── pretrained_models
     │   ├── hrnet_pretrain.pth
     │   └── hrnet_w32_conv_pare_coco.pth
     ├── preprocessed_datasets

diff --git a/configs/pare/hrnet_w32_conv_pare_mix.py b/configs/pare/hrnet_w32_conv_pare_mix.py
@@ -201,11 +201,11 @@
                 ann_file='eft_mpii.npz'),
             dict(
                 type=dataset_type,
-                dataset_name='mpi-inf-3dhp',
+                dataset_name='mpi_inf_3dhp',
                 data_prefix='data',
                 pipeline=train_pipeline,
                 convention='smpl_24',
-                ann_file='mpi_inf_3dhp_train_mmhuman3d.npz'),
+                ann_file='mpi_inf_3dhp_train.npz'),
         ],
         partition=[0.5, 0.233, 0.046, 0.021, 0.2],
     ),

diff --git a/configs/pare/hrnet_w32_conv_pare_mix_cache.py b/configs/pare/hrnet_w32_conv_pare_mix_cache.py
@@ -213,12 +213,12 @@
                 ann_file='eft_mpii.npz'),
             dict(
                 type=dataset_type,
-                dataset_name='mpi-inf-3dhp',
+                dataset_name='mpi_inf_3dhp',
                 data_prefix='data',
                 pipeline=train_pipeline,
                 convention='smpl_24',
                 cache_data_path=cache_files['mpi-inf-3dhp'],
-                ann_file='mpi_inf_3dhp_train_mmhuman3d.npz'),
+                ann_file='mpi_inf_3dhp_train.npz'),
         ],
         partition=[0.5, 0.233, 0.046, 0.021, 0.2],
     ),

diff --git a/configs/pare/hrnet_w32_conv_pare_mix_no_mosh.py b/configs/pare/hrnet_w32_conv_pare_mix_no_mosh.py
@@ -201,11 +201,11 @@
                 ann_file='eft_mpii.npz'),
             dict(
                 type=dataset_type,
-                dataset_name='mpi-inf-3dhp',
+                dataset_name='mpi_inf_3dhp',
                 data_prefix='data',
                 pipeline=train_pipeline,
                 convention='smpl_24',
-                ann_file='mpi_inf_3dhp_train_mmhuman3d.npz'),
+                ann_file='mpi_inf_3dhp_train.npz'),
         ],
         partition=[0.5, 0.233, 0.046, 0.021, 0.2],
     ),