[Fix] Modify configs use "img_res" (#316)

* configs format correction * format correction
open-mmlab · Feb 25, 2023 · 2e54142 · 2e54142
1 parent 0e0c2ce
commit 2e54142
Show file tree

Hide file tree

Showing 17 changed files with 53 additions and 53 deletions.
diff --git a/configs/balanced_mse/resnet50_spin_ihmr_ft_baseline.py b/configs/balanced_mse/resnet50_spin_ihmr_ft_baseline.py
@@ -113,7 +113,7 @@
     dict(type='RandomChannelNoise', noise_factor=0.4),
     dict(type='RandomHorizontalFlip', flip_prob=0.5, convention='smpl_49'),
     dict(type='GetRandomScaleRotation', rot_factor=30, scale_factor=0.25),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -128,7 +128,7 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='GetRandomScaleRotation', rot_factor=0, scale_factor=0),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -139,7 +139,7 @@
 ]
 
 inference_pipeline = [
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(

diff --git a/configs/balanced_mse/resnet50_spin_ihmr_ft_bmc.py b/configs/balanced_mse/resnet50_spin_ihmr_ft_bmc.py
@@ -120,7 +120,7 @@
     dict(type='RandomChannelNoise', noise_factor=0.4),
     dict(type='RandomHorizontalFlip', flip_prob=0.5, convention='smpl_49'),
     dict(type='GetRandomScaleRotation', rot_factor=30, scale_factor=0.25),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -135,7 +135,7 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='GetRandomScaleRotation', rot_factor=0, scale_factor=0),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -146,7 +146,7 @@
 ]
 
 inference_pipeline = [
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(

diff --git a/configs/gta_human/hmr/resnet50_hmr_gta_bt.py b/configs/gta_human/hmr/resnet50_hmr_gta_bt.py
@@ -75,7 +75,7 @@
     dict(type='RandomChannelNoise', noise_factor=0.4),
     dict(type='RandomHorizontalFlip', flip_prob=0.5, convention='smpl_54'),
     dict(type='GetRandomScaleRotation', rot_factor=30, scale_factor=0.25),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -91,7 +91,7 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='GetRandomScaleRotation', rot_factor=0, scale_factor=0),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -102,7 +102,7 @@
 ]
 
 inference_pipeline = [
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(

diff --git a/configs/gta_human/hmr/resnet50_hmr_gta_ft.py b/configs/gta_human/hmr/resnet50_hmr_gta_ft.py
@@ -77,7 +77,7 @@
     dict(type='RandomChannelNoise', noise_factor=0.4),
     dict(type='RandomHorizontalFlip', flip_prob=0.5, convention='smpl_54'),
     dict(type='GetRandomScaleRotation', rot_factor=30, scale_factor=0.25),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -93,7 +93,7 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='GetRandomScaleRotation', rot_factor=0, scale_factor=0),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -104,7 +104,7 @@
 ]
 
 inference_pipeline = [
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(

diff --git a/configs/gta_human/pare/hrnet_w32_conv_pare_gta_ft.py b/configs/gta_human/pare/hrnet_w32_conv_pare_gta_ft.py
@@ -132,7 +132,7 @@
     dict(type='RandomChannelNoise', noise_factor=0.4),
     dict(type='RandomHorizontalFlip', flip_prob=0.5, convention='smpl_24'),
     dict(type='GetRandomScaleRotation', rot_factor=30, scale_factor=0.25),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -145,7 +145,7 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='GetRandomScaleRotation', rot_factor=0, scale_factor=0),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -156,7 +156,7 @@
 ]
 
 inference_pipeline = [
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(

diff --git a/configs/gta_human/spin/resnet50_spin_gta_ft.py b/configs/gta_human/spin/resnet50_spin_gta_ft.py
@@ -108,7 +108,7 @@
     dict(type='RandomChannelNoise', noise_factor=0.4),
     dict(type='RandomHorizontalFlip', flip_prob=0.5, convention='smpl_49'),
     dict(type='GetRandomScaleRotation', rot_factor=30, scale_factor=0.25),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -118,7 +118,7 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='GetRandomScaleRotation', rot_factor=0, scale_factor=0),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -129,7 +129,7 @@
 ]
 
 inference_pipeline = [
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(

diff --git a/configs/hmr/resnet50_hmr_pw3d.py b/configs/hmr/resnet50_hmr_pw3d.py
@@ -75,7 +75,7 @@
     dict(type='RandomChannelNoise', noise_factor=0.4),
     dict(type='RandomHorizontalFlip', flip_prob=0.5, convention='smpl_54'),
     dict(type='GetRandomScaleRotation', rot_factor=30, scale_factor=0.25),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -91,7 +91,7 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='GetRandomScaleRotation', rot_factor=0, scale_factor=0),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -102,7 +102,7 @@
 ]
 
 inference_pipeline = [
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(

diff --git a/configs/hmr/resnet50_hmr_pw3d_e50_cache.py b/configs/hmr/resnet50_hmr_pw3d_e50_cache.py
@@ -75,7 +75,7 @@
     dict(type='RandomChannelNoise', noise_factor=0.4),
     dict(type='RandomHorizontalFlip', flip_prob=0.5, convention='smpl_54'),
     dict(type='GetRandomScaleRotation', rot_factor=30, scale_factor=0.25),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -91,7 +91,7 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='GetRandomScaleRotation', rot_factor=0, scale_factor=0),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -102,7 +102,7 @@
 ]
 
 inference_pipeline = [
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(

diff --git a/configs/hybrik/resnet34_hybrik_mixed.py b/configs/hybrik/resnet34_hybrik_mixed.py
@@ -93,8 +93,8 @@
     dict(type='RandomOcclusion', occlusion_prob=0.9),
     dict(type='HybrIKRandomFlip', flip_prob=0.5, flip_pairs=flip_pairs),
     dict(type='NewKeypointsSelection', maps=keypoints_maps),
-    dict(type='HybrIKAffine', img_res=256),
-    dict(type='GenerateHybrIKTarget', img_res=256, test_mode=False),
+    dict(type='HybrIKAffine', img_res=img_res),
+    dict(type='GenerateHybrIKTarget', img_res=img_res, test_mode=False),
     dict(type='RandomChannelNoise', noise_factor=0.4),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
@@ -108,8 +108,8 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='NewKeypointsSelection', maps=keypoints_maps),
-    dict(type='HybrIKAffine', img_res=256),
-    dict(type='GenerateHybrIKTarget', img_res=256, test_mode=True),
+    dict(type='HybrIKAffine', img_res=img_res),
+    dict(type='GenerateHybrIKTarget', img_res=img_res, test_mode=True),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(
@@ -130,8 +130,8 @@
 test_hp3d_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='NewKeypointsSelection', maps=hp3d_keypoints_map),
-    dict(type='HybrIKAffine', img_res=256),
-    dict(type='GenerateHybrIKTarget', img_res=256, test_mode=True),
+    dict(type='HybrIKAffine', img_res=img_res),
+    dict(type='GenerateHybrIKTarget', img_res=img_res, test_mode=True),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(

diff --git a/configs/pare/hrnet_w32_conv_pare_coco.py b/configs/pare/hrnet_w32_conv_pare_coco.py
@@ -135,7 +135,7 @@
         occluders_file='data/occluders/pascal_occluders.npy'),
     dict(type='RandomHorizontalFlip', flip_prob=0.5, convention='smpl_49'),
     dict(type='GetRandomScaleRotation', rot_factor=30, scale_factor=0.25),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -148,7 +148,7 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='GetRandomScaleRotation', rot_factor=0, scale_factor=0),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -159,7 +159,7 @@
 ]
 
 inference_pipeline = [
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(

diff --git a/configs/pare/hrnet_w32_conv_pare_coco_cache.py b/configs/pare/hrnet_w32_conv_pare_coco_cache.py
@@ -135,7 +135,7 @@
         occluders_file='data/occluders/pascal_occluders.npy'),
     dict(type='RandomHorizontalFlip', flip_prob=0.5, convention='smpl_49'),
     dict(type='GetRandomScaleRotation', rot_factor=30, scale_factor=0.25),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -148,7 +148,7 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='GetRandomScaleRotation', rot_factor=0, scale_factor=0),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -159,7 +159,7 @@
 ]
 
 inference_pipeline = [
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(

diff --git a/configs/pare/hrnet_w32_conv_pare_mix.py b/configs/pare/hrnet_w32_conv_pare_mix.py
@@ -132,7 +132,7 @@
     dict(type='RandomChannelNoise', noise_factor=0.4),
     dict(type='RandomHorizontalFlip', flip_prob=0.5, convention='smpl_24'),
     dict(type='GetRandomScaleRotation', rot_factor=30, scale_factor=0.25),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -145,7 +145,7 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='GetRandomScaleRotation', rot_factor=0, scale_factor=0),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -156,7 +156,7 @@
 ]
 
 inference_pipeline = [
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(

diff --git a/configs/pare/hrnet_w32_conv_pare_mix_cache.py b/configs/pare/hrnet_w32_conv_pare_mix_cache.py
@@ -132,7 +132,7 @@
     dict(type='RandomChannelNoise', noise_factor=0.4),
     dict(type='RandomHorizontalFlip', flip_prob=0.5, convention='smpl_24'),
     dict(type='GetRandomScaleRotation', rot_factor=30, scale_factor=0.25),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -145,7 +145,7 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='GetRandomScaleRotation', rot_factor=0, scale_factor=0),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -156,7 +156,7 @@
 ]
 
 inference_pipeline = [
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(

diff --git a/configs/pare/hrnet_w32_conv_pare_mix_no_mosh.py b/configs/pare/hrnet_w32_conv_pare_mix_no_mosh.py
@@ -132,7 +132,7 @@
     dict(type='RandomChannelNoise', noise_factor=0.4),
     dict(type='RandomHorizontalFlip', flip_prob=0.5, convention='smpl_24'),
     dict(type='GetRandomScaleRotation', rot_factor=30, scale_factor=0.25),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -145,7 +145,7 @@
 test_pipeline = [
     dict(type='LoadImageFromFile'),
     dict(type='GetRandomScaleRotation', rot_factor=0, scale_factor=0),
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(type='ToTensor', keys=data_keys),
@@ -156,7 +156,7 @@
 ]
 
 inference_pipeline = [
-    dict(type='MeshAffine', img_res=224),
+    dict(type='MeshAffine', img_res=img_res),
     dict(type='Normalize', **img_norm_cfg),
     dict(type='ImageToTensor', keys=['img']),
     dict(