diff --git a/configs/_base_/datasets/300w.py b/configs/_base_/datasets/300w.py index 277d00a720..10c343a2ad 100644 --- a/configs/_base_/datasets/300w.py +++ b/configs/_base_/datasets/300w.py @@ -382,23 +382,3 @@ skeleton_info={}, joint_weights=[1.] * 68, sigmas=[]) - -data_root = 'data/300w' - -data = dict( - train=dict( - type='Face300WDataset', - ann_file=f'{data_root}/annotations/face_landmarks_300w_train.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - val=dict( - type='Face300WDataset', - ann_file=f'{data_root}/annotations/face_landmarks_300w_valid.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='Face300WDataset', - ann_file=f'{data_root}/annotations/face_landmarks_300w_valid.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/aflw.py b/configs/_base_/datasets/aflw.py index e011089c83..bf534cbb75 100644 --- a/configs/_base_/datasets/aflw.py +++ b/configs/_base_/datasets/aflw.py @@ -81,23 +81,3 @@ skeleton_info={}, joint_weights=[1.] * 19, sigmas=[]) - -data_root = 'data/aflw' - -data = dict( - train=dict( - type='FaceAFLWDataset', - ann_file=f'{data_root}/annotations/face_landmarks_aflw_train.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - val=dict( - type='FaceAFLWDataset', - ann_file=f'{data_root}/annotations/face_landmarks_aflw_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='FaceAFLWDataset', - ann_file=f'{data_root}/annotations/face_landmarks_aflw_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/aic.py b/configs/_base_/datasets/aic.py index bc1f62e86d..9ecdbe3f0a 100644 --- a/configs/_base_/datasets/aic.py +++ b/configs/_base_/datasets/aic.py @@ -138,26 +138,3 @@ 0.03909642, 0.03686941, 0.01981803, 0.03843971, 0.03412318, 0.02415081, 0.01291456, 0.01236173 ]) - -data_root = 'data/aic' - -data = dict( - train=dict( - type='TopDownAicDataset', - ann_file=f'{data_root}/annotations/aic_train.json', - img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' - 'keypoint_train_images_20170902/', - dataset_info=dataset_info), - val=dict( - type='TopDownAicDataset', - ann_file=f'{data_root}/annotations/aic_val.json', - img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' - 'keypoint_validation_images_20170911/', - dataset_info=dataset_info), - test=dict( - type='TopDownAicDataset', - ann_file=f'{data_root}/annotations/aic_val.json', - img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' - 'keypoint_validation_images_20170911/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/animalpose.py b/configs/_base_/datasets/animalpose.py index ee0192e9d4..d5bb62d951 100644 --- a/configs/_base_/datasets/animalpose.py +++ b/configs/_base_/datasets/animalpose.py @@ -164,23 +164,3 @@ 0.025, 0.025, 0.026, 0.035, 0.035, 0.10, 0.10, 0.10, 0.107, 0.107, 0.107, 0.107, 0.087, 0.087, 0.087, 0.087, 0.089, 0.089, 0.089, 0.089 ]) - -data_root = 'data/animalpose' - -data = dict( - train=dict( - type='AnimalPoseDataset', - ann_file=f'{data_root}/annotations/animalpose_train.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - val=dict( - type='AnimalPoseDataset', - ann_file=f'{data_root}/annotations/animalpose_val.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - test=dict( - type='AnimalPoseDataset', - ann_file=f'{data_root}/annotations/animalpose_val.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/atrw.py b/configs/_base_/datasets/atrw.py index 2fd8b8dd3d..7ec71c8c50 100644 --- a/configs/_base_/datasets/atrw.py +++ b/configs/_base_/datasets/atrw.py @@ -142,23 +142,3 @@ 0.0277, 0.0823, 0.0831, 0.0202, 0.0716, 0.0263, 0.0646, 0.0302, 0.0440, 0.0316, 0.0333, 0.0547, 0.0263, 0.0683, 0.0539 ]) - -data_root = 'data/atrw' - -data = dict( - train=dict( - type='AnimalATRWDataset', - ann_file=f'{data_root}/annotations/keypoint_train.json', - img_prefix=f'{data_root}/images/train/', - dataset_info=dataset_info), - val=dict( - type='AnimalATRWDataset', - ann_file=f'{data_root}/annotations/keypoint_val.json', - img_prefix=f'{data_root}/images/val/', - dataset_info=dataset_info), - test=dict( - type='AnimalATRWDataset', - ann_file=f'{data_root}/annotations/keypoint_val.json', - img_prefix=f'{data_root}/images/val/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/coco.py b/configs/_base_/datasets/coco.py index 5908fa52c4..865a95bc02 100644 --- a/configs/_base_/datasets/coco.py +++ b/configs/_base_/datasets/coco.py @@ -179,23 +179,3 @@ 0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062, 0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089 ]) - -data_root = 'data/coco' - -data = dict( - train=dict( - type='TopDownCocoDataset', - ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', - img_prefix=f'{data_root}/train2017/', - dataset_info=dataset_info), - val=dict( - type='TopDownCocoDataset', - ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', - img_prefix=f'{data_root}/val2017/', - dataset_info=dataset_info), - test=dict( - type='TopDownCocoDataset', - ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', - img_prefix=f'{data_root}/val2017/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/coco_wholebody.py b/configs/_base_/datasets/coco_wholebody.py index 3bdbc8d625..1756862b54 100644 --- a/configs/_base_/datasets/coco_wholebody.py +++ b/configs/_base_/datasets/coco_wholebody.py @@ -1148,23 +1148,3 @@ 0.035, 0.018, 0.024, 0.022, 0.026, 0.017, 0.021, 0.021, 0.032, 0.02, 0.019, 0.022, 0.031 ]) - -data_root = 'data/coco' - -data = dict( - train=dict( - type='TopDownCocoWholeBodyDataset', - ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', - img_prefix=f'{data_root}/train2017/', - dataset_info=dataset_info), - val=dict( - type='TopDownCocoWholeBodyDataset', - ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', - img_prefix=f'{data_root}/val2017/', - dataset_info=dataset_info), - test=dict( - type='TopDownCocoWholeBodyDataset', - ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', - img_prefix=f'{data_root}/val2017/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/cofw.py b/configs/_base_/datasets/cofw.py index 01699f7200..2fb7ad2f8d 100644 --- a/configs/_base_/datasets/cofw.py +++ b/configs/_base_/datasets/cofw.py @@ -132,23 +132,3 @@ skeleton_info={}, joint_weights=[1.] * 29, sigmas=[]) - -data_root = 'data/cofw' - -data = dict( - train=dict( - type='FaceCOFWDataset', - ann_file=f'{data_root}/annotations/cofw_train.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - val=dict( - type='FaceCOFWDataset', - ann_file=f'{data_root}/annotations/cofw_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='FaceCOFWDataset', - ann_file=f'{data_root}/annotations/cofw_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/crowdpose.py b/configs/_base_/datasets/crowdpose.py index fd12f0a14c..45086531a6 100644 --- a/configs/_base_/datasets/crowdpose.py +++ b/configs/_base_/datasets/crowdpose.py @@ -145,23 +145,3 @@ 0.079, 0.079, 0.072, 0.072, 0.062, 0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089, 0.079, 0.079 ]) - -data_root = 'data/crowdpose' - -data = dict( - train=dict( - type='TopDownCrowdPoseDataset', - ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - val=dict( - type='TopDownCrowdPoseDataset', - ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='TopDownCrowdPoseDataset', - ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/deepfashion_full.py b/configs/_base_/datasets/deepfashion_full.py index 5cf329ceab..4d989069ee 100644 --- a/configs/_base_/datasets/deepfashion_full.py +++ b/configs/_base_/datasets/deepfashion_full.py @@ -72,23 +72,3 @@ skeleton_info={}, joint_weights=[1.] * 8, sigmas=[]) - -data_root = 'data/fld' - -data = dict( - train=dict( - type='DeepFashionDataset', - ann_file=f'{data_root}/annotations/fld_full_train.json', - img_prefix=f'{data_root}/img/', - dataset_info=dataset_info), - val=dict( - type='DeepFashionDataset', - ann_file=f'{data_root}/annotations/fld_full_val.json', - img_prefix=f'{data_root}/img/', - dataset_info=dataset_info), - test=dict( - type='DeepFashionDataset', - ann_file=f'{data_root}/annotations/fld_full_test.json', - img_prefix=f'{data_root}/img/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/deepfashion_lower.py b/configs/_base_/datasets/deepfashion_lower.py index 54e3eb9caa..db014a1747 100644 --- a/configs/_base_/datasets/deepfashion_lower.py +++ b/configs/_base_/datasets/deepfashion_lower.py @@ -44,23 +44,3 @@ skeleton_info={}, joint_weights=[1.] * 4, sigmas=[]) - -data_root = 'data/fld' - -data = dict( - train=dict( - type='DeepFashionDataset', - ann_file=f'{data_root}/annotations/fld_lower_train.json', - img_prefix=f'{data_root}/img/', - dataset_info=dataset_info), - val=dict( - type='DeepFashionDataset', - ann_file=f'{data_root}/annotations/fld_lower_val.json', - img_prefix=f'{data_root}/img/', - dataset_info=dataset_info), - test=dict( - type='DeepFashionDataset', - ann_file=f'{data_root}/annotations/fld_lower_test.json', - img_prefix=f'{data_root}/img/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/deepfashion_upper.py b/configs/_base_/datasets/deepfashion_upper.py index 385fc4bbe4..f0b012fd37 100644 --- a/configs/_base_/datasets/deepfashion_upper.py +++ b/configs/_base_/datasets/deepfashion_upper.py @@ -58,23 +58,3 @@ skeleton_info={}, joint_weights=[1.] * 6, sigmas=[]) - -data_root = 'data/fld' - -data = dict( - train=dict( - type='DeepFashionDataset', - ann_file=f'{data_root}/annotations/fld_upper_train.json', - img_prefix=f'{data_root}/img/', - dataset_info=dataset_info), - val=dict( - type='DeepFashionDataset', - ann_file=f'{data_root}/annotations/fld_upper_val.json', - img_prefix=f'{data_root}/img/', - dataset_info=dataset_info), - test=dict( - type='DeepFashionDataset', - ann_file=f'{data_root}/annotations/fld_upper_test.json', - img_prefix=f'{data_root}/img/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/fly.py b/configs/_base_/datasets/fly.py index 74dad0be57..5f94ff57ca 100644 --- a/configs/_base_/datasets/fly.py +++ b/configs/_base_/datasets/fly.py @@ -235,23 +235,3 @@ }, joint_weights=[1.] * 32, sigmas=[]) - -data_root = 'data/fly' - -data = dict( - train=dict( - type='AnimalFlyDataset', - ann_file=f'{data_root}/annotations/fly_train.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - val=dict( - type='AnimalFlyDataset', - ann_file=f'{data_root}/annotations/fly_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='AnimalFlyDataset', - ann_file=f'{data_root}/annotations/fly_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/freihand2d.py b/configs/_base_/datasets/freihand2d.py index ca6b5acf1a..8b960d10f3 100644 --- a/configs/_base_/datasets/freihand2d.py +++ b/configs/_base_/datasets/freihand2d.py @@ -142,23 +142,3 @@ }, joint_weights=[1.] * 21, sigmas=[]) - -data_root = 'data/freihand' - -data = dict( - train=dict( - type='FreiHandDataset', - ann_file=f'{data_root}/annotations/freihand_train.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - val=dict( - type='FreiHandDataset', - ann_file=f'{data_root}/annotations/freihand_val.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - test=dict( - type='FreiHandDataset', - ann_file=f'{data_root}/annotations/freihand_test.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/h36m.py b/configs/_base_/datasets/h36m.py index 7a647367f8..68b7e50ad3 100644 --- a/configs/_base_/datasets/h36m.py +++ b/configs/_base_/datasets/h36m.py @@ -149,26 +149,3 @@ }, joint_weights=[1.] * 17, sigmas=[]) - -data_root = 'data/h36m' -data = dict( - samples_per_gpu=64, - workers_per_gpu=2, - val_dataloader=dict(samples_per_gpu=64), - test_dataloader=dict(samples_per_gpu=64), - train=dict( - type='Body3DH36MDataset', - ann_file=f'{data_root}/annotation_body3d/fps50/h36m_train.npz', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - val=dict( - type='Body3DH36MDataset', - ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='Body3DH36MDataset', - ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/horse10.py b/configs/_base_/datasets/horse10.py index 0ec4757e3d..a485bf191b 100644 --- a/configs/_base_/datasets/horse10.py +++ b/configs/_base_/datasets/horse10.py @@ -199,23 +199,3 @@ }, joint_weights=[1.] * 22, sigmas=[]) - -data_root = 'data/horse10' - -data = dict( - train=dict( - type='AnimalHorse10Dataset', - ann_file=f'{data_root}/annotations/horse10-train-split1.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - val=dict( - type='AnimalHorse10Dataset', - ann_file=f'{data_root}/annotations/horse10-test-split1.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - test=dict( - type='AnimalHorse10Dataset', - ann_file=f'{data_root}/annotations/horse10-test-split1.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/interhand2d.py b/configs/_base_/datasets/interhand2d.py index f3351f1373..0134f07de5 100644 --- a/configs/_base_/datasets/interhand2d.py +++ b/configs/_base_/datasets/interhand2d.py @@ -140,38 +140,3 @@ }, joint_weights=[1.] * 21, sigmas=[]) - -data_root = 'data/interhand2.6m' - -data = dict( - train=dict( - type='InterHand2DDataset', - ann_file=f'{data_root}/annotations/all/' - 'InterHand2.6M_train_data.json', - camera_file=f'{data_root}/annotations/all/' - 'InterHand2.6M_train_camera.json', - joint_file=f'{data_root}/annotations/all/' - 'InterHand2.6M_train_joint_3d.json', - img_prefix=f'{data_root}/images/train/', - dataset_info=dataset_info), - val=dict( - type='InterHand2DDataset', - ann_file=f'{data_root}/annotations/machine_annot/' - 'InterHand2.6M_val_data.json', - camera_file=f'{data_root}/annotations/machine_annot/' - 'InterHand2.6M_val_camera.json', - joint_file=f'{data_root}/annotations/machine_annot/' - 'InterHand2.6M_val_joint_3d.json', - img_prefix=f'{data_root}/images/val/', - dataset_info=dataset_info), - test=dict( - type='InterHand2DDataset', - ann_file=f'{data_root}/annotations/all/' - 'InterHand2.6M_test_data.json', - camera_file=f'{data_root}/annotations/all/' - 'InterHand2.6M_test_camera.json', - joint_file=f'{data_root}/annotations/all/' - 'InterHand2.6M_test_joint_3d.json', - img_prefix=f'{data_root}/images/test/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/interhand3d.py b/configs/_base_/datasets/interhand3d.py index ac9c2e427b..e2bd8121c2 100644 --- a/configs/_base_/datasets/interhand3d.py +++ b/configs/_base_/datasets/interhand3d.py @@ -485,38 +485,3 @@ }, joint_weights=[1.] * 42, sigmas=[]) - -data_root = 'data/interhand2.6m' - -data = dict( - train=dict( - type='InterHand3DDataset', - ann_file=f'{data_root}/annotations/all/' - 'InterHand2.6M_train_data.json', - camera_file=f'{data_root}/annotations/all/' - 'InterHand2.6M_train_camera.json', - joint_file=f'{data_root}/annotations/all/' - 'InterHand2.6M_train_joint_3d.json', - img_prefix=f'{data_root}/images/train/', - dataset_info=dataset_info), - val=dict( - type='InterHand3DDataset', - ann_file=f'{data_root}/annotations/machine_annot/' - 'InterHand2.6M_val_data.json', - camera_file=f'{data_root}/annotations/machine_annot/' - 'InterHand2.6M_val_camera.json', - joint_file=f'{data_root}/annotations/machine_annot/' - 'InterHand2.6M_val_joint_3d.json', - img_prefix=f'{data_root}/images/val/', - dataset_info=dataset_info), - test=dict( - type='InterHand3DDataset', - ann_file=f'{data_root}/annotations/all/' - 'InterHand2.6M_test_data.json', - camera_file=f'{data_root}/annotations/all/' - 'InterHand2.6M_test_camera.json', - joint_file=f'{data_root}/annotations/all/' - 'InterHand2.6M_test_joint_3d.json', - img_prefix=f'{data_root}/images/test/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/jhmdb.py b/configs/_base_/datasets/jhmdb.py index 36f0d03c7a..1b37488498 100644 --- a/configs/_base_/datasets/jhmdb.py +++ b/configs/_base_/datasets/jhmdb.py @@ -127,23 +127,3 @@ 0.025, 0.107, 0.025, 0.079, 0.079, 0.107, 0.107, 0.072, 0.072, 0.087, 0.087, 0.062, 0.062, 0.089, 0.089 ]) - -data_root = 'data/jhmdb' - -data = dict( - train=dict( - type='TopDownJhmdbDataset', - ann_file=f'{data_root}/annotations/Sub1_train.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - val=dict( - type='TopDownJhmdbDataset', - ann_file=f'{data_root}/annotations/Sub1_test.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - test=dict( - type='TopDownJhmdbDataset', - ann_file=f'{data_root}/annotations/Sub1_test.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/locust.py b/configs/_base_/datasets/locust.py index 09670b608d..db3fa15aa0 100644 --- a/configs/_base_/datasets/locust.py +++ b/configs/_base_/datasets/locust.py @@ -261,23 +261,3 @@ }, joint_weights=[1.] * 35, sigmas=[]) - -data_root = 'data/locust' - -data = dict( - train=dict( - type='AnimalLocustDataset', - ann_file=f'{data_root}/annotations/locust_train.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - val=dict( - type='AnimalLocustDataset', - ann_file=f'{data_root}/annotations/locust_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='AnimalLocustDataset', - ann_file=f'{data_root}/annotations/locust_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/macaque.py b/configs/_base_/datasets/macaque.py index 4f43e55c85..12d2480fd1 100644 --- a/configs/_base_/datasets/macaque.py +++ b/configs/_base_/datasets/macaque.py @@ -181,23 +181,3 @@ 0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062, 0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089 ]) - -data_root = 'data/macaque' - -data = dict( - train=dict( - type='AnimalMacaqueDataset', - ann_file=f'{data_root}/annotations/macaque_train.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - val=dict( - type='AnimalMacaqueDataset', - ann_file=f'{data_root}/annotations/macaque_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='AnimalMacaqueDataset', - ann_file=f'{data_root}/annotations/macaque_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/mhp.py b/configs/_base_/datasets/mhp.py index 09e58f6634..e16e37c79c 100644 --- a/configs/_base_/datasets/mhp.py +++ b/configs/_base_/datasets/mhp.py @@ -154,23 +154,3 @@ 0.089, 0.083, 0.107, 0.107, 0.083, 0.089, 0.026, 0.026, 0.026, 0.026, 0.062, 0.072, 0.179, 0.179, 0.072, 0.062 ]) - -data_root = 'data/mpii' - -data = dict( - train=dict( - type='TopDownMpiiDataset', - ann_file=f'{data_root}/annotations/mpii_train.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - val=dict( - type='TopDownMpiiDataset', - ann_file=f'{data_root}/annotations/mpii_val.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='TopDownMpiiDataset', - ann_file=f'{data_root}/annotations/mpii_val.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/mpii.py b/configs/_base_/datasets/mpii.py index d09d6ca598..6c2a491c7b 100644 --- a/configs/_base_/datasets/mpii.py +++ b/configs/_base_/datasets/mpii.py @@ -153,23 +153,3 @@ 0.089, 0.083, 0.107, 0.107, 0.083, 0.089, 0.026, 0.026, 0.026, 0.026, 0.062, 0.072, 0.179, 0.179, 0.072, 0.062 ]) - -data_root = 'data/mpii' - -data = dict( - train=dict( - type='TopDownMpiiDataset', - ann_file=f'{data_root}/annotations/mpii_train.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - val=dict( - type='TopDownMpiiDataset', - ann_file=f'{data_root}/annotations/mpii_val.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='TopDownMpiiDataset', - ann_file=f'{data_root}/annotations/mpii_val.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/mpii_trb.py b/configs/_base_/datasets/mpii_trb.py index ba34c029ae..73940d4b48 100644 --- a/configs/_base_/datasets/mpii_trb.py +++ b/configs/_base_/datasets/mpii_trb.py @@ -378,23 +378,3 @@ }, joint_weights=[1.] * 40, sigmas=[]) - -data_root = 'data/mpii' - -data = dict( - train=dict( - type='TopDownMpiiTrbDataset', - ann_file=f'{data_root}/annotations/mpii_trb_train.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - val=dict( - type='TopDownMpiiTrbDataset', - ann_file=f'{data_root}/annotations/mpii_trb_val.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='TopDownMpiiTrbDataset', - ann_file=f'{data_root}/annotations/mpii_trb_val.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/ochuman.py b/configs/_base_/datasets/ochuman.py index f2eff65ab8..2ef20838fe 100644 --- a/configs/_base_/datasets/ochuman.py +++ b/configs/_base_/datasets/ochuman.py @@ -179,25 +179,3 @@ 0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062, 0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089 ]) - -data_root = 'data/ochuman' - -data = dict( - train=dict( - type='TopDownCocoDataset', - ann_file='data/coco/annotations/person_keypoints_train2017.json', - img_prefix='data/coco//train2017/', - dataset_info=dataset_info), - val=dict( - type='TopDownOCHumanDataset', - ann_file=f'{data_root}/annotations/' - 'ochuman_coco_format_val_range_0.00_1.00.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='TopDownOCHumanDataset', - ann_file=f'{data_root}/annotations/' - 'ochuman_coco_format_test_range_0.00_1.00.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/onehand10k.py b/configs/_base_/datasets/onehand10k.py index ffd6ce96d9..016770f14f 100644 --- a/configs/_base_/datasets/onehand10k.py +++ b/configs/_base_/datasets/onehand10k.py @@ -140,23 +140,3 @@ }, joint_weights=[1.] * 21, sigmas=[]) - -data_root = 'data/onehand10k' - -data = dict( - train=dict( - type='OneHand10KDataset', - ann_file=f'{data_root}/annotations/onehand10k_train.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - val=dict( - type='OneHand10KDataset', - ann_file=f'{data_root}/annotations/onehand10k_test.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - test=dict( - type='OneHand10KDataset', - ann_file=f'{data_root}/annotations/onehand10k_test.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/panoptic_hand2d.py b/configs/_base_/datasets/panoptic_hand2d.py index 33745826b5..7a65731ba8 100644 --- a/configs/_base_/datasets/panoptic_hand2d.py +++ b/configs/_base_/datasets/panoptic_hand2d.py @@ -141,23 +141,3 @@ }, joint_weights=[1.] * 21, sigmas=[]) - -data_root = 'data/panoptic' - -data = dict( - train=dict( - type='PanopticDataset', - ann_file=f'{data_root}/annotations/panoptic_train.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - val=dict( - type='PanopticDataset', - ann_file=f'{data_root}/annotations/panoptic_test.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - test=dict( - type='PanopticDataset', - ann_file=f'{data_root}/annotations/panoptic_test.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/posetrack18.py b/configs/_base_/datasets/posetrack18.py index 1300754bbb..5aefd1c97f 100644 --- a/configs/_base_/datasets/posetrack18.py +++ b/configs/_base_/datasets/posetrack18.py @@ -174,23 +174,3 @@ 0.026, 0.025, 0.025, 0.035, 0.035, 0.079, 0.079, 0.072, 0.072, 0.062, 0.062, 0.107, 0.107, 0.087, 0.087, 0.089, 0.089 ]) - -data_root = 'data/posetrack18' - -data = dict( - train=dict( - type='TopDownPoseTrack18Dataset', - ann_file=f'{data_root}/annotations/posetrack18_train.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - val=dict( - type='TopDownPoseTrack18Dataset', - ann_file=f'{data_root}/annotations/posetrack18_val.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - test=dict( - type='TopDownPoseTrack18Dataset', - ann_file=f'{data_root}/annotations/posetrack18_val.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/rhd2d.py b/configs/_base_/datasets/rhd2d.py index c2721f0e49..f48e637026 100644 --- a/configs/_base_/datasets/rhd2d.py +++ b/configs/_base_/datasets/rhd2d.py @@ -139,23 +139,3 @@ }, joint_weights=[1.] * 21, sigmas=[]) - -data_root = 'data/rhd' - -data = dict( - train=dict( - type='Rhd2DDataset', - ann_file=f'{data_root}/annotations/rhd_train.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - val=dict( - type='Rhd2DDataset', - ann_file=f'{data_root}/annotations/rhd_test.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), - test=dict( - type='Rhd2DDataset', - ann_file=f'{data_root}/annotations/rhd_test.json', - img_prefix=f'{data_root}/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/wflw.py b/configs/_base_/datasets/wflw.py index fdfc56c5c1..bed6f56f30 100644 --- a/configs/_base_/datasets/wflw.py +++ b/configs/_base_/datasets/wflw.py @@ -580,23 +580,3 @@ skeleton_info={}, joint_weights=[1.] * 98, sigmas=[]) - -data_root = 'data/wflw' - -data = dict( - train=dict( - type='FaceWFLWDataset', - ann_file=f'{data_root}/annotations/face_landmarks_wflw_train.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - val=dict( - type='FaceWFLWDataset', - ann_file=f'{data_root}/annotations/face_landmarks_wflw_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='FaceWFLWDataset', - ann_file=f'{data_root}/annotations/face_landmarks_wflw_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/_base_/datasets/zebra.py b/configs/_base_/datasets/zebra.py index cb6ef5f479..eac71f796a 100644 --- a/configs/_base_/datasets/zebra.py +++ b/configs/_base_/datasets/zebra.py @@ -62,23 +62,3 @@ }, joint_weights=[1.] * 9, sigmas=[]) - -data_root = 'data/zebra' - -data = dict( - train=dict( - type='AnimalZebraDataset', - ann_file=f'{data_root}/annotations/zebra_train.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - val=dict( - type='AnimalZebraDataset', - ann_file=f'{data_root}/annotations/zebra_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), - test=dict( - type='AnimalZebraDataset', - ann_file=f'{data_root}/annotations/zebra_test.json', - img_prefix=f'{data_root}/images/', - dataset_info=dataset_info), -) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_w32_animalpose_256x256.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_w32_animalpose_256x256.py index c7f88005a0..96df6a384b 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_w32_animalpose_256x256.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_w32_animalpose_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/animalpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -155,17 +156,20 @@ ann_file=f'{data_root}/annotations/animalpose_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalPoseDataset', ann_file=f'{data_root}/annotations/animalpose_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalPoseDataset', ann_file=f'{data_root}/annotations/animalpose_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_w48_animalpose_256x256.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_w48_animalpose_256x256.py index fe16347b24..d92d1ecab8 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_w48_animalpose_256x256.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/hrnet_w48_animalpose_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/animalpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -155,17 +156,20 @@ ann_file=f'{data_root}/annotations/animalpose_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalPoseDataset', ann_file=f'{data_root}/annotations/animalpose_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalPoseDataset', ann_file=f'{data_root}/annotations/animalpose_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res101_animalpose_256x256.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res101_animalpose_256x256.py index e0f3faa277..b6f989724a 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res101_animalpose_256x256.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res101_animalpose_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/animalpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -124,17 +125,20 @@ ann_file=f'{data_root}/annotations/animalpose_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalPoseDataset', ann_file=f'{data_root}/annotations/animalpose_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalPoseDataset', ann_file=f'{data_root}/annotations/animalpose_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res152_animalpose_256x256.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res152_animalpose_256x256.py index e0e6e7e834..011eb04510 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res152_animalpose_256x256.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res152_animalpose_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/animalpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -124,17 +125,20 @@ ann_file=f'{data_root}/annotations/animalpose_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalPoseDataset', ann_file=f'{data_root}/annotations/animalpose_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalPoseDataset', ann_file=f'{data_root}/annotations/animalpose_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res50_animalpose_256x256.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res50_animalpose_256x256.py index 3687ea88fb..3141703243 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res50_animalpose_256x256.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/animalpose/res50_animalpose_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/animalpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -124,17 +125,20 @@ ann_file=f'{data_root}/annotations/animalpose_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalPoseDataset', ann_file=f'{data_root}/annotations/animalpose_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalPoseDataset', ann_file=f'{data_root}/annotations/animalpose_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_w32_atrw_256x256.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_w32_atrw_256x256.py index b6692f08b8..a9b8301fd5 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_w32_atrw_256x256.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_w32_atrw_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/atrw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -153,17 +154,20 @@ ann_file=f'{data_root}/annotations/keypoint_train.json', img_prefix=f'{data_root}/images/train/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalATRWDataset', ann_file=f'{data_root}/annotations/keypoint_val.json', img_prefix=f'{data_root}/images/val/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalATRWDataset', ann_file=f'{data_root}/annotations/keypoint_val.json', img_prefix=f'{data_root}/images/val/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_w48_atrw_256x256.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_w48_atrw_256x256.py index 38da4b63b5..019010e713 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_w48_atrw_256x256.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/hrnet_w48_atrw_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/atrw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -153,17 +154,20 @@ ann_file=f'{data_root}/annotations/keypoint_train.json', img_prefix=f'{data_root}/images/train/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalATRWDataset', ann_file=f'{data_root}/annotations/keypoint_val.json', img_prefix=f'{data_root}/images/val/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalATRWDataset', ann_file=f'{data_root}/annotations/keypoint_val.json', img_prefix=f'{data_root}/images/val/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res101_atrw_256x256.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res101_atrw_256x256.py index 6055fd5219..4a428da2d5 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res101_atrw_256x256.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res101_atrw_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/atrw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,17 +123,20 @@ ann_file=f'{data_root}/annotations/keypoint_train.json', img_prefix=f'{data_root}/images/train/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalATRWDataset', ann_file=f'{data_root}/annotations/keypoint_val.json', img_prefix=f'{data_root}/images/val/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalATRWDataset', ann_file=f'{data_root}/annotations/keypoint_val.json', img_prefix=f'{data_root}/images/val/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res152_atrw_256x256.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res152_atrw_256x256.py index 8f39958e11..65a1892871 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res152_atrw_256x256.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res152_atrw_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/atrw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,17 +123,20 @@ ann_file=f'{data_root}/annotations/keypoint_train.json', img_prefix=f'{data_root}/images/train/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalATRWDataset', ann_file=f'{data_root}/annotations/keypoint_val.json', img_prefix=f'{data_root}/images/val/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalATRWDataset', ann_file=f'{data_root}/annotations/keypoint_val.json', img_prefix=f'{data_root}/images/val/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res50_atrw_256x256.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res50_atrw_256x256.py index 89a6c52083..f16d8b5096 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res50_atrw_256x256.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/atrw/res50_atrw_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/atrw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,17 +123,20 @@ ann_file=f'{data_root}/annotations/keypoint_train.json', img_prefix=f'{data_root}/images/train/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalATRWDataset', ann_file=f'{data_root}/annotations/keypoint_val.json', img_prefix=f'{data_root}/images/val/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalATRWDataset', ann_file=f'{data_root}/annotations/keypoint_val.json', img_prefix=f'{data_root}/images/val/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/fly/res101_fly_192x192.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/fly/res101_fly_192x192.py index 00ed5c06c9..4860a21c47 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/fly/res101_fly_192x192.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/fly/res101_fly_192x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/fly.py'] log_level = 'INFO' load_from = None resume_from = None @@ -114,17 +115,20 @@ ann_file=f'{data_root}/annotations/fly_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalFlyDataset', ann_file=f'{data_root}/annotations/fly_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalFlyDataset', ann_file=f'{data_root}/annotations/fly_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/fly/res152_fly_192x192.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/fly/res152_fly_192x192.py index 9d494eb40a..97697ccf28 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/fly/res152_fly_192x192.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/fly/res152_fly_192x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/fly.py'] log_level = 'INFO' load_from = None resume_from = None @@ -114,17 +115,20 @@ ann_file=f'{data_root}/annotations/fly_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalFlyDataset', ann_file=f'{data_root}/annotations/fly_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalFlyDataset', ann_file=f'{data_root}/annotations/fly_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/fly/res50_fly_192x192.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/fly/res50_fly_192x192.py index 03017f7acf..b6143c57a2 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/fly/res50_fly_192x192.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/fly/res50_fly_192x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/fly.py'] log_level = 'INFO' load_from = None resume_from = None @@ -114,17 +115,20 @@ ann_file=f'{data_root}/annotations/fly_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalFlyDataset', ann_file=f'{data_root}/annotations/fly_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalFlyDataset', ann_file=f'{data_root}/annotations/fly_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w32_horse10_256x256-split1.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w32_horse10_256x256-split1.py index c08fa706bc..a5b721d2c5 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w32_horse10_256x256-split1.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w32_horse10_256x256-split1.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -75,7 +76,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -147,17 +148,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w32_horse10_256x256-split2.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w32_horse10_256x256-split2.py index 913ce21acb..e21e58b342 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w32_horse10_256x256-split2.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w32_horse10_256x256-split2.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -75,7 +76,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -147,17 +148,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w32_horse10_256x256-split3.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w32_horse10_256x256-split3.py index a7e9e44fad..cf04601bcf 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w32_horse10_256x256-split3.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w32_horse10_256x256-split3.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -75,7 +76,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -147,17 +148,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w48_horse10_256x256-split1.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w48_horse10_256x256-split1.py index ed15b0c28a..81b498b351 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w48_horse10_256x256-split1.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w48_horse10_256x256-split1.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -75,7 +76,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -147,17 +148,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w48_horse10_256x256-split2.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w48_horse10_256x256-split2.py index 31849d4d61..8a0ec19eb4 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w48_horse10_256x256-split2.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w48_horse10_256x256-split2.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -147,17 +148,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w48_horse10_256x256-split3.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w48_horse10_256x256-split3.py index 298993e7c7..8e78ac6d2f 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w48_horse10_256x256-split3.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/hrnet_w48_horse10_256x256-split3.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -147,17 +148,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res101_horse10_256x256-split1.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res101_horse10_256x256-split1.py index c737ca8d87..0e01308f87 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res101_horse10_256x256-split1.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res101_horse10_256x256-split1.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -46,7 +47,7 @@ pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -116,17 +117,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res101_horse10_256x256-split2.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res101_horse10_256x256-split2.py index 94f0b7b64d..53672b5860 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res101_horse10_256x256-split2.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res101_horse10_256x256-split2.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -46,7 +47,7 @@ pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -116,17 +117,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res101_horse10_256x256-split3.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res101_horse10_256x256-split3.py index ec147da207..5729660f92 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res101_horse10_256x256-split3.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res101_horse10_256x256-split3.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -116,17 +117,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res152_horse10_256x256-split1.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res152_horse10_256x256-split1.py index 3a9d092972..293cdb170d 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res152_horse10_256x256-split1.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res152_horse10_256x256-split1.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -116,17 +117,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res152_horse10_256x256-split2.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res152_horse10_256x256-split2.py index 61aac58a22..a90e1314e5 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res152_horse10_256x256-split2.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res152_horse10_256x256-split2.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -116,17 +117,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res152_horse10_256x256-split3.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res152_horse10_256x256-split3.py index 63757288db..7c6f91e936 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res152_horse10_256x256-split3.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res152_horse10_256x256-split3.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -116,17 +117,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res50_horse10_256x256-split1.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res50_horse10_256x256-split1.py index 02be5a65a4..c2da2cf12e 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res50_horse10_256x256-split1.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res50_horse10_256x256-split1.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -46,7 +47,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -116,17 +117,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split1.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res50_horse10_256x256-split2.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res50_horse10_256x256-split2.py index d3a9dd9332..0059a1f706 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res50_horse10_256x256-split2.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res50_horse10_256x256-split2.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -46,7 +47,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -116,17 +117,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split2.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res50_horse10_256x256-split3.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res50_horse10_256x256-split3.py index 0e5bcafa49..c6d870c188 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res50_horse10_256x256-split3.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/horse10/res50_horse10_256x256-split3.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/horse10.py'] log_level = 'INFO' load_from = None resume_from = None @@ -116,17 +117,20 @@ ann_file=f'{data_root}/annotations/horse10-train-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalHorse10Dataset', ann_file=f'{data_root}/annotations/horse10-test-split3.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/locust/res101_locust_160x160.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/locust/res101_locust_160x160.py index 870682082f..7fcfb42a4d 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/locust/res101_locust_160x160.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/locust/res101_locust_160x160.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/locust.py'] log_level = 'INFO' load_from = None resume_from = None @@ -114,17 +115,20 @@ ann_file=f'{data_root}/annotations/locust_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalLocustDataset', ann_file=f'{data_root}/annotations/locust_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalLocustDataset', ann_file=f'{data_root}/annotations/locust_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/locust/res152_locust_160x160.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/locust/res152_locust_160x160.py index 01ee823326..ea299622f5 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/locust/res152_locust_160x160.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/locust/res152_locust_160x160.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/locust.py'] log_level = 'INFO' load_from = None resume_from = None @@ -114,17 +115,20 @@ ann_file=f'{data_root}/annotations/locust_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalLocustDataset', ann_file=f'{data_root}/annotations/locust_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalLocustDataset', ann_file=f'{data_root}/annotations/locust_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/locust/res50_locust_160x160.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/locust/res50_locust_160x160.py index af917d7051..95d067f22c 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/locust/res50_locust_160x160.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/locust/res50_locust_160x160.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/locust.py'] log_level = 'INFO' load_from = None resume_from = None @@ -114,17 +115,20 @@ ann_file=f'{data_root}/annotations/locust_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalLocustDataset', ann_file=f'{data_root}/annotations/locust_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalLocustDataset', ann_file=f'{data_root}/annotations/locust_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/hrnet_w32_macaque_256x192.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/hrnet_w32_macaque_256x192.py index 2219c74b74..57c790db58 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/hrnet_w32_macaque_256x192.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/hrnet_w32_macaque_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/macaque.py'] log_level = 'INFO' load_from = None resume_from = None @@ -155,17 +156,20 @@ ann_file=f'{data_root}/annotations/macaque_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalMacaqueDataset', ann_file=f'{data_root}/annotations/macaque_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalMacaqueDataset', ann_file=f'{data_root}/annotations/macaque_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/hrnet_w48_macaque_256x192.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/hrnet_w48_macaque_256x192.py index 7b5628cefd..7700162cfa 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/hrnet_w48_macaque_256x192.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/hrnet_w48_macaque_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/macaque.py'] log_level = 'INFO' load_from = None resume_from = None @@ -155,17 +156,20 @@ ann_file=f'{data_root}/annotations/macaque_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalMacaqueDataset', ann_file=f'{data_root}/annotations/macaque_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalMacaqueDataset', ann_file=f'{data_root}/annotations/macaque_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/res101_macaque_256x192.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/res101_macaque_256x192.py index 7c9bf2134e..37c84e3185 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/res101_macaque_256x192.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/res101_macaque_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/macaque.py'] log_level = 'INFO' load_from = None resume_from = None @@ -124,17 +125,20 @@ ann_file=f'{data_root}/annotations/macaque_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalMacaqueDataset', ann_file=f'{data_root}/annotations/macaque_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalMacaqueDataset', ann_file=f'{data_root}/annotations/macaque_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/res152_macaque_256x192.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/res152_macaque_256x192.py index ba273ed548..9b845d66e1 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/res152_macaque_256x192.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/res152_macaque_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/macaque.py'] log_level = 'INFO' load_from = None resume_from = None @@ -124,17 +125,20 @@ ann_file=f'{data_root}/annotations/macaque_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalMacaqueDataset', ann_file=f'{data_root}/annotations/macaque_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalMacaqueDataset', ann_file=f'{data_root}/annotations/macaque_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/res50_macaque_256x192.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/res50_macaque_256x192.py index bebb9d80e3..96405baf99 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/res50_macaque_256x192.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/macaque/res50_macaque_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/macaque.py'] log_level = 'INFO' load_from = None resume_from = None @@ -124,17 +125,20 @@ ann_file=f'{data_root}/annotations/macaque_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalMacaqueDataset', ann_file=f'{data_root}/annotations/macaque_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalMacaqueDataset', ann_file=f'{data_root}/annotations/macaque_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/zebra/res101_zebra_160x160.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/zebra/res101_zebra_160x160.py index 7cb9820682..ed1c7ac2c2 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/zebra/res101_zebra_160x160.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/zebra/res101_zebra_160x160.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/zebra.py'] log_level = 'INFO' load_from = None resume_from = None @@ -108,17 +109,20 @@ ann_file=f'{data_root}/annotations/zebra_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalZebraDataset', ann_file=f'{data_root}/annotations/zebra_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalZebraDataset', ann_file=f'{data_root}/annotations/zebra_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/zebra/res152_zebra_160x160.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/zebra/res152_zebra_160x160.py index 40ca6bf6fc..5c2d985b55 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/zebra/res152_zebra_160x160.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/zebra/res152_zebra_160x160.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/zebra.py'] log_level = 'INFO' load_from = None resume_from = None @@ -108,17 +109,20 @@ ann_file=f'{data_root}/annotations/zebra_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalZebraDataset', ann_file=f'{data_root}/annotations/zebra_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalZebraDataset', ann_file=f'{data_root}/annotations/zebra_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/zebra/res50_zebra_160x160.py b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/zebra/res50_zebra_160x160.py index c9d0a9ce48..af17ce333f 100644 --- a/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/zebra/res50_zebra_160x160.py +++ b/configs/animal/2d_kpt_sview_rgb_img/topdown_heatmap/zebra/res50_zebra_160x160.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/zebra.py'] log_level = 'INFO' load_from = None resume_from = None @@ -108,17 +109,20 @@ ann_file=f'{data_root}/annotations/zebra_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='AnimalZebraDataset', ann_file=f'{data_root}/annotations/zebra_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='AnimalZebraDataset', ann_file=f'{data_root}/annotations/zebra_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/aic/higherhrnet_w32_aic_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/aic/higherhrnet_w32_aic_512x512.py index 18232da1a6..0d57d9d0af 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/aic/higherhrnet_w32_aic_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/aic/higherhrnet_w32_aic_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aic.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=32, num_joints=14, tag_per_joint=True, @@ -181,19 +182,22 @@ img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' 'keypoint_train_images_20170902/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/aic/higherhrnet_w32_aic_512x512_udp.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/aic/higherhrnet_w32_aic_512x512_udp.py index b4ab95a9f7..f1b8d75b6a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/aic/higherhrnet_w32_aic_512x512_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/aic/higherhrnet_w32_aic_512x512_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aic.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=32, num_joints=14, tag_per_joint=True, @@ -185,19 +186,22 @@ img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' 'keypoint_train_images_20170902/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/aic/hrnet_w32_aic_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/aic/hrnet_w32_aic_512x512.py index b47d60d2d5..8e4369193f 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/aic/hrnet_w32_aic_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/aic/hrnet_w32_aic_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aic.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=32, num_joints=14, num_deconv_layers=0, @@ -177,19 +178,22 @@ img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' 'keypoint_train_images_20170902/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_512x512.py index c90d963323..7bac396105 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=32, num_joints=17, tag_per_joint=True, @@ -181,17 +182,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_512x512_udp.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_512x512_udp.py index fce8a984e4..85636eb7cc 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_512x512_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_512x512_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=32, num_joints=17, tag_per_joint=True, @@ -185,17 +186,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_640x640.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_640x640.py index be70c08c8f..ee35766b4d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_640x640.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_640x640.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=32, num_joints=17, tag_per_joint=True, @@ -181,17 +182,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_640x640_udp.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_640x640_udp.py index a188c49658..75b0ebe3e4 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_640x640_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w32_coco_640x640_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=32, num_joints=17, tag_per_joint=True, @@ -185,17 +186,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w48_coco_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w48_coco_512x512.py index 9c2556dcea..45d391ee7c 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w48_coco_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w48_coco_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w48-8ef0771d.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=48, num_joints=17, tag_per_joint=True, @@ -181,17 +182,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w48_coco_512x512_udp.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w48_coco_512x512_udp.py index 21e4e9e676..ae1d2acc1e 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w48_coco_512x512_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/higherhrnet_w48_coco_512x512_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w48-8ef0771d.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=48, num_joints=17, tag_per_joint=True, @@ -185,17 +186,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_512x512.py index af264c5d80..da9d341c68 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=32, num_joints=17, num_deconv_layers=0, @@ -177,17 +178,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_512x512_udp.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_512x512_udp.py index fcce049a87..554f7db237 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_512x512_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_512x512_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=32, num_joints=17, num_deconv_layers=0, @@ -181,17 +182,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_640x640.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_640x640.py index 3932853586..19237e9254 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_640x640.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_640x640.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=32, num_joints=17, num_deconv_layers=0, @@ -177,17 +178,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_640x640_udp.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_640x640_udp.py index d7bd592ddf..f8c2042605 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_640x640_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w32_coco_640x640_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=32, num_joints=17, num_deconv_layers=0, @@ -181,17 +182,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_512x512.py index 2f1d9c2b4c..aabfe9c08c 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w48-8ef0771d.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=48, num_joints=17, num_deconv_layers=0, @@ -177,17 +178,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_512x512_udp.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_512x512_udp.py index 8be8badf1a..786251651e 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_512x512_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_512x512_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w48-8ef0771d.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=48, num_joints=17, num_deconv_layers=0, @@ -181,17 +182,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_640x640.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_640x640.py index 3696316448..1f7ede93d0 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_640x640.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_640x640.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w48-8ef0771d.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=48, num_joints=17, num_deconv_layers=0, @@ -177,17 +178,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_640x640_udp.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_640x640_udp.py index c1dc7849f9..d4251b3342 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_640x640_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/hrnet_w48_coco_640x640_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,7 +50,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w48-8ef0771d.pth', backbone=dict( @@ -82,7 +83,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=48, num_joints=17, num_deconv_layers=0, @@ -181,17 +182,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/mobilenetv2_coco_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/mobilenetv2_coco_512x512.py index 3f5dc99686..a6bfccdd7c 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/mobilenetv2_coco_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/mobilenetv2_coco_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,11 +50,11 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='mmcls://mobilenet_v2', backbone=dict(type='MobileNetV2', widen_factor=1., out_indices=(7, )), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=1280, num_joints=17, tag_per_joint=True, @@ -146,17 +147,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res101_coco_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res101_coco_512x512.py index 025daa8266..fcccd0eeb5 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res101_coco_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res101_coco_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,11 +50,11 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=2048, num_joints=17, tag_per_joint=True, @@ -146,17 +147,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res101_coco_640x640.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res101_coco_640x640.py index 8d0ae90a12..85a03c6f7a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res101_coco_640x640.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res101_coco_640x640.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,11 +50,11 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=2048, num_joints=17, tag_per_joint=True, @@ -146,17 +147,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res152_coco_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res152_coco_512x512.py index 4cb0aeb580..84fd9e2c1c 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res152_coco_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res152_coco_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,11 +50,11 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=2048, num_joints=17, tag_per_joint=True, @@ -146,17 +147,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res152_coco_640x640.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res152_coco_640x640.py index d75f69e511..0bf31d647f 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res152_coco_640x640.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res152_coco_640x640.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,11 +50,11 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=2048, num_joints=17, tag_per_joint=True, @@ -146,17 +147,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res50_coco_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res50_coco_512x512.py index 058cac49ef..5a41d789b4 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res50_coco_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res50_coco_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,11 +50,11 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=2048, num_joints=17, tag_per_joint=True, @@ -147,17 +148,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res50_coco_640x640.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res50_coco_640x640.py index 68e1716705..2e1cc56daa 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res50_coco_640x640.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/coco/res50_coco_640x640.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -49,11 +50,11 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=2048, num_joints=17, tag_per_joint=True, @@ -146,17 +147,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_512x512.py index 4746ae594f..d1521d5f1f 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=32, num_joints=14, tag_per_joint=True, @@ -180,17 +181,20 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_512x512_udp.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_512x512_udp.py index 7538300d2c..b824fb8645 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_512x512_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_512x512_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=32, num_joints=14, tag_per_joint=True, @@ -184,17 +185,20 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_640x640.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_640x640.py index ece8ccbb9b..0b6352eaa1 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_640x640.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_640x640.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=32, num_joints=14, tag_per_joint=True, @@ -180,17 +181,20 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_640x640_udp.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_640x640_udp.py index 24c33b3d60..e9e53a697b 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_640x640_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w32_crowdpose_640x640_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=32, num_joints=14, tag_per_joint=True, @@ -184,17 +185,20 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w48_crowdpose_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w48_crowdpose_512x512.py index 8addf87d6c..388da22b13 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w48_crowdpose_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w48_crowdpose_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w48-8ef0771d.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=48, num_joints=14, tag_per_joint=True, @@ -180,17 +181,20 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w48_crowdpose_512x512_udp.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w48_crowdpose_512x512_udp.py index 0ee83bba23..49847e2386 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w48_crowdpose_512x512_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/higherhrnet_w48_crowdpose_512x512_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w48-8ef0771d.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=48, num_joints=14, tag_per_joint=True, @@ -184,17 +185,20 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/mobilenetv2_crowdpose_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/mobilenetv2_crowdpose_512x512.py index 2f035e99e6..8dc67fe2e5 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/mobilenetv2_crowdpose_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/mobilenetv2_crowdpose_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,11 +49,11 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='mmcls://mobilenet_v2', backbone=dict(type='MobileNetV2', widen_factor=1., out_indices=(7, )), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=1280, num_joints=14, tag_per_joint=True, @@ -145,17 +146,20 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/res101_crowdpose_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/res101_crowdpose_512x512.py index 8fb3959670..3d68902fa0 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/res101_crowdpose_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/res101_crowdpose_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,11 +49,11 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=2048, num_joints=14, tag_per_joint=True, @@ -145,17 +146,20 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/res152_crowdpose_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/res152_crowdpose_512x512.py index 45d740e699..dae1b7474d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/res152_crowdpose_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/res152_crowdpose_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,11 +49,11 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=2048, num_joints=14, tag_per_joint=True, @@ -145,17 +146,20 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/res50_crowdpose_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/res50_crowdpose_512x512.py index aa161e3fe2..9515eefcf5 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/res50_crowdpose_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/crowdpose/res50_crowdpose_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,11 +49,11 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=2048, num_joints=14, tag_per_joint=True, @@ -146,17 +147,20 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/mhp/hrnet_w48_mhp_512x512.py b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/mhp/hrnet_w48_mhp_512x512.py index 27ce967f17..633d8a2de6 100644 --- a/configs/body/2d_kpt_sview_rgb_img/associative_embedding/mhp/hrnet_w48_mhp_512x512.py +++ b/configs/body/2d_kpt_sview_rgb_img/associative_embedding/mhp/hrnet_w48_mhp_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mhp.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w48-8ef0771d.pth', backbone=dict( @@ -80,7 +81,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=48, num_joints=16, num_deconv_layers=0, @@ -175,17 +176,20 @@ ann_file=f'{data_root}/annotations/mhp_train.json', img_prefix=f'{data_root}/train/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpMhpDataset', ann_file=f'{data_root}/annotations/mhp_val.json', img_prefix=f'{data_root}/val/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpMhpDataset', ann_file=f'{data_root}/annotations/mhp_val.json', img_prefix=f'{data_root}/val/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/deeppose/coco/res101_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/deeppose/coco/res101_coco_256x192.py index ab316a4f5b..9d25750261 100644 --- a/configs/body/2d_kpt_sview_rgb_img/deeppose/coco/res101_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/deeppose/coco/res101_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -43,7 +44,7 @@ backbone=dict(type='ResNet', depth=101, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -122,17 +123,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/deeppose/coco/res152_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/deeppose/coco/res152_coco_256x192.py index c864da7c9b..c9c7305040 100644 --- a/configs/body/2d_kpt_sview_rgb_img/deeppose/coco/res152_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/deeppose/coco/res152_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -43,7 +44,7 @@ backbone=dict(type='ResNet', depth=152, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -122,17 +123,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/deeppose/coco/res50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/deeppose/coco/res50_coco_256x192.py index 9694d59176..af5e16802a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/deeppose/coco/res50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/deeppose/coco/res50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -43,7 +44,7 @@ backbone=dict(type='ResNet', depth=50, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -122,17 +123,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/deeppose/mpii/res101_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/deeppose/mpii/res101_mpii_256x256.py index 80f25cfa30..bd5c34556d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/deeppose/mpii/res101_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/deeppose/mpii/res101_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -37,7 +38,7 @@ backbone=dict(type='ResNet', depth=101, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -101,17 +102,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/deeppose/mpii/res152_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/deeppose/mpii/res152_mpii_256x256.py index 617e686a2a..180c9fed27 100644 --- a/configs/body/2d_kpt_sview_rgb_img/deeppose/mpii/res152_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/deeppose/mpii/res152_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -37,7 +38,7 @@ backbone=dict(type='ResNet', depth=152, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -101,17 +102,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/deeppose/mpii/res50_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/deeppose/mpii/res50_mpii_256x256.py index 9ca4ec5e40..1f74698019 100644 --- a/configs/body/2d_kpt_sview_rgb_img/deeppose/mpii/res50_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/deeppose/mpii/res50_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -37,7 +38,7 @@ backbone=dict(type='ResNet', depth=50, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -101,17 +102,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w32_aic_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w32_aic_256x192.py index c5520f570d..d065b96b36 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w32_aic_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w32_aic_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aic.py'] log_level = 'INFO' load_from = None resume_from = None @@ -153,19 +154,22 @@ img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' 'keypoint_train_images_20170902/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w32_aic_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w32_aic_384x288.py index 0dd3025be6..64692b28b5 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w32_aic_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w32_aic_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aic.py'] log_level = 'INFO' load_from = None resume_from = None @@ -153,19 +154,22 @@ img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' 'keypoint_train_images_20170902/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w48_aic_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w48_aic_256x192.py index c71a3c5ae4..3c1c3b5c67 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w48_aic_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w48_aic_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aic.py'] log_level = 'INFO' load_from = None resume_from = None @@ -153,19 +154,22 @@ img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' 'keypoint_train_images_20170902/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w48_aic_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w48_aic_384x288.py index d339e01f23..0000a7cc0e 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w48_aic_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/hrnet_w48_aic_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aic.py'] log_level = 'INFO' load_from = None resume_from = None @@ -154,19 +155,22 @@ img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' 'keypoint_train_images_20170902/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res101_aic_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res101_aic_256x192.py index c07b6d06a3..c7854cac8d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res101_aic_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res101_aic_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aic.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,19 +123,22 @@ img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' 'keypoint_train_images_20170902/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res101_aic_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res101_aic_384x288.py index 1451fef046..34f1d4541a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res101_aic_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res101_aic_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aic.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,19 +123,22 @@ img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' 'keypoint_train_images_20170902/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res152_aic_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res152_aic_256x192.py index 6d8545fa2c..477bf5a756 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res152_aic_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res152_aic_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aic.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,19 +123,22 @@ img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' 'keypoint_train_images_20170902/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res152_aic_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res152_aic_384x288.py index 36b67fab19..44411b2d59 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res152_aic_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res152_aic_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aic.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,19 +123,22 @@ img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' 'keypoint_train_images_20170902/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res50_aic_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res50_aic_256x192.py index 5502ed33fb..7f585d8722 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res50_aic_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res50_aic_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aic.py'] log_level = 'INFO' load_from = None resume_from = None @@ -121,19 +122,22 @@ img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' 'keypoint_train_images_20170902/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res50_aic_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res50_aic_384x288.py index f76bb6ac63..9be230dd4d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res50_aic_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/aic/res50_aic_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aic.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,19 +123,22 @@ img_prefix=f'{data_root}/ai_challenger_keypoint_train_20170902/' 'keypoint_train_images_20170902/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownAicDataset', ann_file=f'{data_root}/annotations/aic_val.json', img_prefix=f'{data_root}/ai_challenger_keypoint_validation_20170911/' 'keypoint_validation_images_20170911/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/2xmspn50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/2xmspn50_coco_256x192.py index e976c566c8..ca4a778bfa 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/2xmspn50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/2xmspn50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ num_blocks=[3, 4, 6, 3], norm_cfg=dict(type='BN')), keypoint_head=dict( - type='TopDownMSMUHead', + type='TopdownHeatmapMSMUHead', out_shape=(64, 48), unit_channels=256, out_channels=channel_cfg['num_output_channels'], @@ -148,17 +149,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/2xrsn50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/2xrsn50_coco_256x192.py index 5cb49b470f..582883466f 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/2xrsn50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/2xrsn50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ num_steps=4, norm_cfg=dict(type='BN')), keypoint_head=dict( - type='TopDownMSMUHead', + type='TopdownHeatmapMSMUHead', out_shape=(64, 48), unit_channels=256, out_channels=channel_cfg['num_output_channels'], @@ -148,17 +149,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/3xmspn50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/3xmspn50_coco_256x192.py index 9f466efc08..b901cf8b49 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/3xmspn50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/3xmspn50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ num_blocks=[3, 4, 6, 3], norm_cfg=dict(type='BN')), keypoint_head=dict( - type='TopDownMSMUHead', + type='TopdownHeatmapMSMUHead', out_shape=(64, 48), unit_channels=256, out_channels=channel_cfg['num_output_channels'], @@ -148,17 +149,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/3xrsn50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/3xrsn50_coco_256x192.py index 5de40f8ba2..375aa80a79 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/3xrsn50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/3xrsn50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ num_steps=4, norm_cfg=dict(type='BN')), keypoint_head=dict( - type='TopDownMSMUHead', + type='TopdownHeatmapMSMUHead', out_shape=(64, 48), unit_channels=256, out_channels=channel_cfg['num_output_channels'], @@ -148,17 +149,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/4xmspn50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/4xmspn50_coco_256x192.py index 771310c54e..8060e5f45d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/4xmspn50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/4xmspn50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ num_blocks=[3, 4, 6, 3], norm_cfg=dict(type='BN')), keypoint_head=dict( - type='TopDownMSMUHead', + type='TopdownHeatmapMSMUHead', out_shape=(64, 48), unit_channels=256, out_channels=channel_cfg['num_output_channels'], @@ -148,17 +149,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/alexnet_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/alexnet_coco_256x192.py index 154b329daf..d1e2b5c281 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/alexnet_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/alexnet_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/cpm_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/cpm_coco_256x192.py index 215423c164..8325d1c067 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/cpm_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/cpm_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ feat_channels=128, num_stages=6), keypoint_head=dict( - type='TopDownMultiStageHead', + type='TopdownHeatmapMultiStageHead', in_channels=channel_cfg['num_output_channels'], out_channels=channel_cfg['num_output_channels'], num_stages=6, @@ -131,17 +132,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/cpm_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/cpm_coco_384x288.py index 26a056cbee..87ac8f14e4 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/cpm_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/cpm_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ feat_channels=128, num_stages=6), keypoint_head=dict( - type='TopDownMultiStageHead', + type='TopdownHeatmapMultiStageHead', in_channels=channel_cfg['num_output_channels'], out_channels=channel_cfg['num_output_channels'], num_stages=6, @@ -131,17 +132,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hourglass52_coco_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hourglass52_coco_256x256.py index 7115a36f6b..1126842a53 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hourglass52_coco_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hourglass52_coco_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -45,7 +46,7 @@ num_stacks=1, ), keypoint_head=dict( - type='TopDownMultiStageHead', + type='TopdownHeatmapMultiStageHead', in_channels=256, out_channels=channel_cfg['num_output_channels'], num_stages=1, @@ -129,17 +130,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hourglass52_coco_384x384.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hourglass52_coco_384x384.py index 1ab426440e..9c030b381c 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hourglass52_coco_384x384.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hourglass52_coco_384x384.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -45,7 +46,7 @@ num_stacks=1, ), keypoint_head=dict( - type='TopDownMultiStageHead', + type='TopdownHeatmapMultiStageHead', in_channels=256, out_channels=channel_cfg['num_output_channels'], num_stages=1, @@ -129,17 +130,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192.py index ec88e54a79..fb3bbe40ac 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -154,17 +155,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_coarsedropout.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_coarsedropout.py index 4f4da37079..e2cc3be3fb 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_coarsedropout.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_coarsedropout.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -71,7 +72,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -167,17 +168,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_dark.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_dark.py index 65141545b7..c233ef4e19 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_dark.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -154,17 +155,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_gridmask.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_gridmask.py index 66a693f37c..8a3d7f4a8a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_gridmask.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_gridmask.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -71,7 +72,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -164,17 +165,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_photometric.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_photometric.py index 81387878cb..b0158b9cfe 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_photometric.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_photometric.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -155,17 +156,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_udp.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_udp.py index 56298a46e5..0a9ced275e 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -161,17 +162,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_udp_regress.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_udp_regress.py index 8dedea307f..d57e33046c 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_udp_regress.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_256x192_udp_regress.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -159,17 +160,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_384x288.py index 2bfeb882dd..a28029552d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -154,17 +155,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_384x288_dark.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_384x288_dark.py index 701a29dc1d..19cc2327e6 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_384x288_dark.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_384x288_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -154,17 +155,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_384x288_udp.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_384x288_udp.py index 60d3976fb7..2fa62e032a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_384x288_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w32_coco_384x288_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -161,17 +162,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_256x192.py index d288144baa..ca8b90b796 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -71,7 +72,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -154,17 +155,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_256x192_dark.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_256x192_dark.py index 286ec10f4a..8d8809f0b5 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_256x192_dark.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_256x192_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -71,7 +72,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -154,17 +155,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_256x192_udp.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_256x192_udp.py index d01b64f273..1f4d0b056f 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_256x192_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_256x192_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -161,17 +162,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_384x288.py index d8171dd5e7..2ad0194c9e 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -71,7 +72,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -154,17 +155,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_384x288_dark.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_384x288_dark.py index 165a786ec9..ea2c2eb2a3 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_384x288_dark.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_384x288_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -71,7 +72,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -154,17 +155,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_384x288_udp.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_384x288_udp.py index 5b1a46d21c..194e2f8bd7 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_384x288_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/hrnet_w48_coco_384x288_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -161,17 +162,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/mobilenetv2_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/mobilenetv2_coco_256x192.py index cffd8437a9..50052d6203 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/mobilenetv2_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/mobilenetv2_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/mobilenetv2_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/mobilenetv2_coco_384x288.py index 90758dc171..8e1dfd5690 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/mobilenetv2_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/mobilenetv2_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/mspn50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/mspn50_coco_256x192.py index 3e98e0e5e3..f1ec581a8d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/mspn50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/mspn50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ num_blocks=[3, 4, 6, 3], norm_cfg=dict(type='BN')), keypoint_head=dict( - type='TopDownMSMUHead', + type='TopdownHeatmapMSMUHead', out_shape=(64, 48), unit_channels=256, out_channels=channel_cfg['num_output_channels'], @@ -147,17 +148,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_256x192.py index fbc3de733c..eda22a84db 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_256x192_dark.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_256x192_dark.py index a0b9a57311..b08905b1ef 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_256x192_dark.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_256x192_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_384x288.py index 0fff7914c8..654a366366 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_384x288_dark.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_384x288_dark.py index 54ef401b37..6115b29ace 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_384x288_dark.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res101_coco_384x288_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_256x192.py index 4f5a9a4e48..78db56d6f6 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_256x192_dark.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_256x192_dark.py index 10358e616d..20edcc882a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_256x192_dark.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_256x192_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_384x288.py index bc96362ad1..e89b6dc15e 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_384x288_dark.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_384x288_dark.py index 1469a07b23..3fd9c59f5d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_384x288_dark.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res152_coco_384x288_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_256x192.py index 489f764053..a01dbcfa03 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -125,17 +126,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_256x192_dark.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_256x192_dark.py index eda6e751a5..cf8afda6c3 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_256x192_dark.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_256x192_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_384x288.py index 7b91f63abb..48f50904c6 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_384x288_dark.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_384x288_dark.py index 09ed350274..b9093ad2e1 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_384x288_dark.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/res50_coco_384x288_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest101_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest101_coco_256x192.py index 8a1f25a639..5f304ba4ec 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest101_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest101_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest101_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest101_coco_384x288.py index 32e7f585e6..0e97110d69 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest101_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest101_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest200_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest200_coco_256x192.py index fc4a8f9e12..dcd8390b21 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest200_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest200_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest200_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest200_coco_384x288.py index e51c45d914..c7358abf4b 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest200_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest200_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='mmcls://resnest200', backbone=dict(type='ResNeSt', depth=200), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest269_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest269_coco_256x192.py index a8c0a63cdd..7b25257c7d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest269_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest269_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='mmcls://resnest269', backbone=dict(type='ResNeSt', depth=269), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest269_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest269_coco_384x288.py index 5149a3daf2..116794e59f 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest269_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest269_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest50_coco_256x192.py index b98c763197..f73eb9f28d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest50_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest50_coco_384x288.py index 9eafca4c82..289978464e 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest50_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnest50_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='mmcls://resnest50', backbone=dict(type='ResNeSt', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d101_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d101_coco_256x192.py index 2f369d9e5c..39a976218b 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d101_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d101_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='mmcls://resnet101_v1d', backbone=dict(type='ResNetV1d', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d101_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d101_coco_384x288.py index cecde01431..fb1b5bfb5f 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d101_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d101_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d152_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d152_coco_256x192.py index f732b9bf73..bb464d6499 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d152_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d152_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='mmcls://resnet152_v1d', backbone=dict(type='ResNetV1d', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d152_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d152_coco_384x288.py index b9d5f1c936..cb73672540 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d152_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d152_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d50_coco_256x192.py index 29196da1ff..a4239d1a0c 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d50_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d50_coco_384x288.py index 860ba1767e..ff9442967d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d50_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnetv1d50_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext101_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext101_coco_256x192.py index f71adf5f08..3ebefbdeaa 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext101_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext101_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='mmcls://resnext101', backbone=dict(type='ResNeXt', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext101_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext101_coco_384x288.py index ff7169cd8b..765c20152a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext101_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext101_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext152_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext152_coco_256x192.py index f4665222a4..0ac7a77968 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext152_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext152_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext152_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext152_coco_384x288.py index 7f455f7d3a..be40d7c80b 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext152_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext152_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext50_coco_256x192.py index 8adfc31609..dd9270e950 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext50_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext50_coco_384x288.py index 80b2009f86..bb2743d679 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext50_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/resnext50_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/rsn18_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/rsn18_coco_256x192.py index 27c1910420..d1413d19a4 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/rsn18_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/rsn18_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ num_steps=4, norm_cfg=dict(type='BN')), keypoint_head=dict( - type='TopDownMSMUHead', + type='TopdownHeatmapMSMUHead', out_shape=(64, 48), unit_channels=256, out_channels=channel_cfg['num_output_channels'], @@ -147,17 +148,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/rsn50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/rsn50_coco_256x192.py index d968f02f65..c91727d101 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/rsn50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/rsn50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ num_steps=4, norm_cfg=dict(type='BN')), keypoint_head=dict( - type='TopDownMSMUHead', + type='TopdownHeatmapMSMUHead', out_shape=(64, 48), unit_channels=256, out_channels=channel_cfg['num_output_channels'], @@ -147,17 +148,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet101_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet101_coco_256x192.py index 6d880424b2..8b85b40b4f 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet101_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet101_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -124,17 +125,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet101_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet101_coco_384x288.py index 4108bae99d..563d943aeb 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet101_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet101_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -124,17 +125,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet50_coco_256x192.py index dffef8c864..7d07626eca 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -43,7 +44,7 @@ 'pretrain_models/scnet50-7ef0a199.pth', backbone=dict(type='SCNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -124,17 +125,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet50_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet50_coco_384x288.py index 386581af47..d7bcccb03e 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet50_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/scnet50_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -124,17 +125,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet101_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet101_coco_256x192.py index 85a51a04c7..b3ff03864a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet101_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet101_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet101_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet101_coco_384x288.py index b55890946f..47c07814da 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet101_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet101_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet152_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet152_coco_256x192.py index f4e29d7f1f..652153ee00 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet152_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet152_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained=None, backbone=dict(type='SEResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet152_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet152_coco_384x288.py index 0cf2dbafbc..60669a7550 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet152_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet152_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained=None, backbone=dict(type='SEResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet50_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet50_coco_256x192.py index c8573fc63c..2a0192996d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet50_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet50_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='mmcls://se-resnet50', backbone=dict(type='SEResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet50_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet50_coco_384x288.py index 6fa87cc7cc..6cd1ab5644 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet50_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/seresnet50_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv1_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv1_coco_256x192.py index e0afc1f970..9627a0789e 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv1_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv1_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv1_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv1_coco_384x288.py index bde62c2e89..ef81697c3a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv1_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv1_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv2_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv2_coco_256x192.py index b71e221879..e32b34da25 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv2_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv2_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv2_coco_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv2_coco_384x288.py index 506ad743e0..6f861e03aa 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv2_coco_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/shufflenetv2_coco_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/vgg16_bn_coco_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/vgg16_bn_coco_256x192.py index 132397b2b3..4b049e3101 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/vgg16_bn_coco_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/vgg16_bn_coco_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='mmcls://vgg16_bn', backbone=dict(type='VGG', depth=16, norm_cfg=dict(type='BN')), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=512, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,17 +124,20 @@ ann_file=f'{data_root}/annotations/person_keypoints_train2017.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoDataset', ann_file=f'{data_root}/annotations/person_keypoints_val2017.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w32_crowdpose_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w32_crowdpose_256x192.py index 95fdb84074..ca64c4d910 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w32_crowdpose_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w32_crowdpose_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -153,13 +154,15 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w32_crowdpose_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w32_crowdpose_384x288.py index 132f65693d..66b886b7fc 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w32_crowdpose_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w32_crowdpose_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -153,13 +154,15 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w48_crowdpose_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w48_crowdpose_256x192.py index 3e65774aff..2766f7bea8 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w48_crowdpose_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w48_crowdpose_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -153,13 +154,15 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w48_crowdpose_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w48_crowdpose_384x288.py index 6195d140a6..4715119801 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w48_crowdpose_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/hrnet_w48_crowdpose_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -153,13 +154,15 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res101_crowdpose_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res101_crowdpose_256x192.py index 5ee3f6f4f8..d760e5e37a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res101_crowdpose_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res101_crowdpose_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,13 +123,15 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res101_crowdpose_320x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res101_crowdpose_320x256.py index 0867d97561..b90f330c3f 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res101_crowdpose_320x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res101_crowdpose_320x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -40,7 +41,7 @@ pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -122,13 +123,15 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res101_crowdpose_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res101_crowdpose_384x288.py index 13b417e07a..4f16f03425 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res101_crowdpose_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res101_crowdpose_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,13 +123,15 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res152_crowdpose_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res152_crowdpose_256x192.py index 66d15803ab..e8d0ae3e7a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res152_crowdpose_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res152_crowdpose_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,13 +123,15 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res152_crowdpose_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res152_crowdpose_384x288.py index 5b1d708f85..71042adfa9 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res152_crowdpose_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res152_crowdpose_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,13 +123,15 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res50_crowdpose_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res50_crowdpose_256x192.py index 49f9821a4d..ecac2fa7f1 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res50_crowdpose_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res50_crowdpose_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,13 +123,15 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res50_crowdpose_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res50_crowdpose_384x288.py index 75f6ab56fc..dfdea731f7 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res50_crowdpose_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/crowdpose/res50_crowdpose_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/crowdpose.py'] log_level = 'INFO' load_from = None resume_from = None @@ -122,13 +123,15 @@ ann_file=f'{data_root}/annotations/mmpose_crowdpose_trainval.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCrowdPoseDataset', ann_file=f'{data_root}/annotations/mmpose_crowdpose_test.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/h36m/hrnet_w32_h36m_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/h36m/hrnet_w32_h36m_256x256.py index a189edcb39..9b89137583 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/h36m/hrnet_w32_h36m_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/h36m/hrnet_w32_h36m_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/h36m.py'] log_level = 'INFO' load_from = None resume_from = None @@ -71,7 +72,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -145,17 +146,20 @@ ann_file=f'{data_root}/annotation_body2d/h36m_coco_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownH36MDataset', ann_file=f'{data_root}/annotation_body2d/h36m_coco_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownH36MDataset', ann_file=f'{data_root}/annotation_body2d/h36m_coco_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/h36m/hrnet_w48_h36m_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/h36m/hrnet_w48_h36m_256x256.py index 6e599c433a..a1ae6ac930 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/h36m/hrnet_w48_h36m_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/h36m/hrnet_w48_h36m_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/h36m.py'] log_level = 'INFO' load_from = None resume_from = None @@ -71,7 +72,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -145,17 +146,20 @@ ann_file=f'{data_root}/annotation_body2d/h36m_coco_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownH36MDataset', ann_file=f'{data_root}/annotation_body2d/h36m_coco_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownH36MDataset', ann_file=f'{data_root}/annotation_body2d/h36m_coco_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/cpm_jhmdb_sub1_368x368.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/cpm_jhmdb_sub1_368x368.py index 315c7a4d42..dcd8a737fd 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/cpm_jhmdb_sub1_368x368.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/cpm_jhmdb_sub1_368x368.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/jhmdb.py'] log_level = 'INFO' load_from = 'https://download.openmmlab.com/mmpose/top_down/cpm/cpm_mpii_368x368-116e62b8_20200822.pth' # noqa: E501 resume_from = None @@ -45,7 +46,7 @@ feat_channels=128, num_stages=6), keypoint_head=dict( - type='TopDownMultiStageHead', + type='TopdownHeatmapMultiStageHead', in_channels=channel_cfg['num_output_channels'], out_channels=channel_cfg['num_output_channels'], num_stages=6, @@ -126,17 +127,20 @@ ann_file=f'{data_root}/annotations/Sub1_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub1_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub1_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/cpm_jhmdb_sub2_368x368.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/cpm_jhmdb_sub2_368x368.py index 08cb4f647f..92a720d872 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/cpm_jhmdb_sub2_368x368.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/cpm_jhmdb_sub2_368x368.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/jhmdb.py'] log_level = 'INFO' load_from = 'https://download.openmmlab.com/mmpose/top_down/cpm/cpm_mpii_368x368-116e62b8_20200822.pth' # noqa: E501 resume_from = None @@ -45,7 +46,7 @@ feat_channels=128, num_stages=6), keypoint_head=dict( - type='TopDownMultiStageHead', + type='TopdownHeatmapMultiStageHead', in_channels=channel_cfg['num_output_channels'], out_channels=channel_cfg['num_output_channels'], num_stages=6, @@ -126,17 +127,20 @@ ann_file=f'{data_root}/annotations/Sub2_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub2_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub2_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/cpm_jhmdb_sub3_368x368.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/cpm_jhmdb_sub3_368x368.py index c5a037a090..05f4addb26 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/cpm_jhmdb_sub3_368x368.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/cpm_jhmdb_sub3_368x368.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/jhmdb.py'] log_level = 'INFO' load_from = 'https://download.openmmlab.com/mmpose/top_down/cpm/cpm_mpii_368x368-116e62b8_20200822.pth' # noqa: E501 resume_from = None @@ -45,7 +46,7 @@ feat_channels=128, num_stages=6), keypoint_head=dict( - type='TopDownMultiStageHead', + type='TopdownHeatmapMultiStageHead', in_channels=channel_cfg['num_output_channels'], out_channels=channel_cfg['num_output_channels'], num_stages=6, @@ -126,17 +127,20 @@ ann_file=f'{data_root}/annotations/Sub3_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub3_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub3_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_2deconv_jhmdb_sub1_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_2deconv_jhmdb_sub1_256x256.py index 1d23107663..654a1b2f10 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_2deconv_jhmdb_sub1_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_2deconv_jhmdb_sub1_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/jhmdb.py'] log_level = 'INFO' load_from = 'https://download.openmmlab.com/mmpose/top_down/resnet/res50_mpii_256x256-418ffc88_20200812.pth' # noqa: E501 resume_from = None @@ -121,17 +122,20 @@ ann_file=f'{data_root}/annotations/Sub1_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub1_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub1_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_2deconv_jhmdb_sub2_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_2deconv_jhmdb_sub2_256x256.py index 01f4846eba..8ad05a8280 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_2deconv_jhmdb_sub2_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_2deconv_jhmdb_sub2_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/jhmdb.py'] log_level = 'INFO' load_from = 'https://download.openmmlab.com/mmpose/top_down/resnet/res50_mpii_256x256-418ffc88_20200812.pth' # noqa: E501 resume_from = None @@ -121,17 +122,20 @@ ann_file=f'{data_root}/annotations/Sub2_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub2_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub2_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_2deconv_jhmdb_sub3_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_2deconv_jhmdb_sub3_256x256.py index 9f50313185..565536c08b 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_2deconv_jhmdb_sub3_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_2deconv_jhmdb_sub3_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/jhmdb.py'] log_level = 'INFO' load_from = 'https://download.openmmlab.com/mmpose/top_down/resnet/res50_mpii_256x256-418ffc88_20200812.pth' # noqa: E501 resume_from = None @@ -121,17 +122,20 @@ ann_file=f'{data_root}/annotations/Sub3_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub3_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub3_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_jhmdb_sub1_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_jhmdb_sub1_256x256.py index f5b70808ab..e1913d1f2c 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_jhmdb_sub1_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_jhmdb_sub1_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/jhmdb.py'] log_level = 'INFO' load_from = 'https://download.openmmlab.com/mmpose/top_down/resnet/res50_mpii_256x256-418ffc88_20200812.pth' # noqa: E501 resume_from = None @@ -118,17 +119,20 @@ ann_file=f'{data_root}/annotations/Sub1_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub1_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub1_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_jhmdb_sub2_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_jhmdb_sub2_256x256.py index 6652fccff7..6bfc95a06f 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_jhmdb_sub2_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_jhmdb_sub2_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/jhmdb.py'] log_level = 'INFO' load_from = 'https://download.openmmlab.com/mmpose/top_down/resnet/res50_mpii_256x256-418ffc88_20200812.pth' # noqa: E501 resume_from = None @@ -118,17 +119,20 @@ ann_file=f'{data_root}/annotations/Sub2_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub2_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub2_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_jhmdb_sub3_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_jhmdb_sub3_256x256.py index c55b612ab4..697e202b94 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_jhmdb_sub3_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/jhmdb/res50_jhmdb_sub3_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/jhmdb.py'] log_level = 'INFO' load_from = 'https://download.openmmlab.com/mmpose/top_down/resnet/res50_mpii_256x256-418ffc88_20200812.pth' # noqa: E501 resume_from = None @@ -118,17 +119,20 @@ ann_file=f'{data_root}/annotations/Sub3_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub3_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownJhmdbDataset', ann_file=f'{data_root}/annotations/Sub3_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mhp/res50_mhp_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mhp/res50_mhp_256x192.py index ad67089bde..c923a2bb22 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mhp/res50_mhp_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mhp/res50_mhp_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mhp.py'] log_level = 'INFO' load_from = None resume_from = None @@ -121,17 +122,20 @@ ann_file=f'{data_root}/annotations/mhp_train.json', img_prefix=f'{data_root}/train/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMhpDataset', ann_file=f'{data_root}/annotations/mhp_val.json', img_prefix=f'{data_root}/val/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMhpDataset', ann_file=f'{data_root}/annotations/mhp_val.json', img_prefix=f'{data_root}/val/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/cpm_mpii_368x368.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/cpm_mpii_368x368.py index 37cae9ee75..08c8e15f86 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/cpm_mpii_368x368.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/cpm_mpii_368x368.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ feat_channels=128, num_stages=6), keypoint_head=dict( - type='TopDownMultiStageHead', + type='TopdownHeatmapMultiStageHead', in_channels=channel_cfg['num_output_channels'], out_channels=channel_cfg['num_output_channels'], num_stages=6, @@ -113,17 +114,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hourglass52_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hourglass52_mpii_256x256.py index a49dd22137..73a42d7486 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hourglass52_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hourglass52_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -39,7 +40,7 @@ num_stacks=1, ), keypoint_head=dict( - type='TopDownMultiStageHead', + type='TopdownHeatmapMultiStageHead', in_channels=256, out_channels=channel_cfg['num_output_channels'], num_stages=1, @@ -110,17 +111,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hourglass52_mpii_384x384.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hourglass52_mpii_384x384.py index efd3587a5e..9b480caa44 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hourglass52_mpii_384x384.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hourglass52_mpii_384x384.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -39,7 +40,7 @@ num_stacks=1, ), keypoint_head=dict( - type='TopDownMultiStageHead', + type='TopdownHeatmapMultiStageHead', in_channels=256, out_channels=channel_cfg['num_output_channels'], num_stages=1, @@ -110,17 +111,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w32_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w32_mpii_256x256.py index 2430ab5599..785bd14ff5 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w32_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w32_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -135,17 +136,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w32_mpii_256x256_dark.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w32_mpii_256x256_dark.py index 60ac401333..faa6c16216 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w32_mpii_256x256_dark.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w32_mpii_256x256_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -65,7 +66,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -135,17 +136,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w32_mpii_256x256_udp.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w32_mpii_256x256_udp.py index b8533e1501..315729ca1b 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w32_mpii_256x256_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w32_mpii_256x256_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -66,7 +67,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -142,17 +143,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w48_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w48_mpii_256x256.py index 2dd63b54ab..584c7a8fea 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w48_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w48_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -65,7 +66,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -135,17 +136,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w48_mpii_256x256_dark.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w48_mpii_256x256_dark.py index 86533e6592..951481ae17 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w48_mpii_256x256_dark.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w48_mpii_256x256_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -65,7 +66,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -135,17 +136,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w48_mpii_256x256_udp.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w48_mpii_256x256_udp.py index f5b7d378be..dbb76b4045 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w48_mpii_256x256_udp.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/hrnet_w48_mpii_256x256_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -66,7 +67,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -142,17 +143,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/mobilenetv2_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/mobilenetv2_mpii_256x256.py index a70fb2c082..2bee7df354 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/mobilenetv2_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/mobilenetv2_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained='mmcls://mobilenet_v2', backbone=dict(type='MobileNetV2', widen_factor=1., out_indices=(7, )), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=1280, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/res101_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/res101_mpii_256x256.py index e6906d0c94..8c897e6a47 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/res101_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/res101_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/res152_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/res152_mpii_256x256.py index a0d824c681..8013e35994 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/res152_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/res152_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/res50_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/res50_mpii_256x256.py index 749b638f19..bf6e24a546 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/res50_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/res50_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnetv1d101_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnetv1d101_mpii_256x256.py index 50ec61bf83..467c6ba590 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnetv1d101_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnetv1d101_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnetv1d152_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnetv1d152_mpii_256x256.py index f3795ca020..317bbbb218 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnetv1d152_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnetv1d152_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained='mmcls://resnet152_v1d', backbone=dict(type='ResNetV1d', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnetv1d50_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnetv1d50_mpii_256x256.py index cc054bacf1..cc3b126ddb 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnetv1d50_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnetv1d50_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained='mmcls://resnet50_v1d', backbone=dict(type='ResNetV1d', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnext101_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnext101_mpii_256x256.py index 5e07247277..3f42f89498 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnext101_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnext101_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnext152_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnext152_mpii_256x256.py index 7289b14fa6..0ada013113 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnext152_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnext152_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained='mmcls://resnext152', backbone=dict(type='ResNeXt', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnext50_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnext50_mpii_256x256.py index e6b103eb73..caaa5c199a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnext50_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/resnext50_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained='mmcls://resnext50', backbone=dict(type='ResNeXt', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/scnet101_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/scnet101_mpii_256x256.py index c547d20d3d..cde3aae0c3 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/scnet101_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/scnet101_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -37,7 +38,7 @@ 'pretrain_models/scnet101-94250a77.pth', backbone=dict(type='SCNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -105,17 +106,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/scnet50_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/scnet50_mpii_256x256.py index 71a8a4a659..bd82d9146a 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/scnet50_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/scnet50_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -37,7 +38,7 @@ 'pretrain_models/scnet50-7ef0a199.pth', backbone=dict(type='SCNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -105,17 +106,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/seresnet101_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/seresnet101_mpii_256x256.py index 4c37d64f49..de45a0eb23 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/seresnet101_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/seresnet101_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained='mmcls://se-resnet101', backbone=dict(type='SEResNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/seresnet152_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/seresnet152_mpii_256x256.py index 9cd9aec80e..8a1f3b3008 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/seresnet152_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/seresnet152_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained=None, backbone=dict(type='SEResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/seresnet50_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/seresnet50_mpii_256x256.py index acd15caf6b..a7ad24165c 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/seresnet50_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/seresnet50_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained='mmcls://se-resnet50', backbone=dict(type='SEResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/shufflenetv1_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/shufflenetv1_mpii_256x256.py index c27ec3d29c..de39932b06 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/shufflenetv1_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/shufflenetv1_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained='mmcls://shufflenet_v1', backbone=dict(type='ShuffleNetV1', groups=3), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=960, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/shufflenetv2_mpii_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/shufflenetv2_mpii_256x256.py index 639fb5b808..b01a28024e 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/shufflenetv2_mpii_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii/shufflenetv2_mpii_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii.py'] log_level = 'INFO' load_from = None resume_from = None @@ -104,17 +105,20 @@ ann_file=f'{data_root}/annotations/mpii_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiDataset', ann_file=f'{data_root}/annotations/mpii_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii_trb/res101_mpii_trb_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii_trb/res101_mpii_trb_256x256.py index bdc35e8733..1a554ed588 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii_trb/res101_mpii_trb_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii_trb/res101_mpii_trb_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii_trb.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,13 +105,15 @@ ann_file=f'{data_root}/annotations/mpii_trb_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiTrbDataset', ann_file=f'{data_root}/annotations/mpii_trb_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiTrbDataset', ann_file=f'{data_root}/annotations/mpii_trb_val.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii_trb/res152_mpii_trb_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii_trb/res152_mpii_trb_256x256.py index 49091ec8b6..15429ece23 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii_trb/res152_mpii_trb_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii_trb/res152_mpii_trb_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii_trb.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,13 +105,15 @@ ann_file=f'{data_root}/annotations/mpii_trb_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiTrbDataset', ann_file=f'{data_root}/annotations/mpii_trb_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiTrbDataset', ann_file=f'{data_root}/annotations/mpii_trb_val.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii_trb/res50_mpii_trb_256x256.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii_trb/res50_mpii_trb_256x256.py index d2b7c670ee..72d169e79d 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii_trb/res50_mpii_trb_256x256.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/mpii_trb/res50_mpii_trb_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/mpii_trb.py'] log_level = 'INFO' load_from = None resume_from = None @@ -36,7 +37,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -104,13 +105,15 @@ ann_file=f'{data_root}/annotations/mpii_trb_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownMpiiTrbDataset', ann_file=f'{data_root}/annotations/mpii_trb_val.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownMpiiTrbDataset', ann_file=f'{data_root}/annotations/mpii_trb_val.json', diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res101_ochuman_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res101_ochuman_256x192.py index e18a68aade..387bf7766f 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res101_ochuman_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res101_ochuman_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/ochuman.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,19 +124,22 @@ ann_file='data/coco/annotations/person_keypoints_train2017.json', img_prefix='data/coco//train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownOCHumanDataset', ann_file=f'{data_root}/annotations/' 'ochuman_coco_format_val_range_0.00_1.00.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownOCHumanDataset', ann_file=f'{data_root}/annotations/' 'ochuman_coco_format_test_range_0.00_1.00.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res101_ochuman_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res101_ochuman_384x288.py index 8d9feb9239..2a68eb47bc 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res101_ochuman_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res101_ochuman_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/ochuman.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,19 +124,22 @@ ann_file='data/coco/annotations/person_keypoints_train2017.json', img_prefix='data/coco//train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownOCHumanDataset', ann_file=f'{data_root}/annotations/' 'ochuman_coco_format_val_range_0.00_1.00.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownOCHumanDataset', ann_file=f'{data_root}/annotations/' 'ochuman_coco_format_test_range_0.00_1.00.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res152_ochuman_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res152_ochuman_256x192.py index 992d762cc3..8ec47f9ad5 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res152_ochuman_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res152_ochuman_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/ochuman.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,19 +124,22 @@ ann_file='data/coco/annotations/person_keypoints_train2017.json', img_prefix='data/coco//train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownOCHumanDataset', ann_file=f'{data_root}/annotations/' 'ochuman_coco_format_val_range_0.00_1.00.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownOCHumanDataset', ann_file=f'{data_root}/annotations/' 'ochuman_coco_format_test_range_0.00_1.00.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res152_ochuman_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res152_ochuman_384x288.py index 0b807a2983..aeb443a025 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res152_ochuman_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res152_ochuman_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/ochuman.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,19 +124,22 @@ ann_file='data/coco/annotations/person_keypoints_train2017.json', img_prefix='data/coco//train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownOCHumanDataset', ann_file=f'{data_root}/annotations/' 'ochuman_coco_format_val_range_0.00_1.00.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownOCHumanDataset', ann_file=f'{data_root}/annotations/' 'ochuman_coco_format_test_range_0.00_1.00.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res50_ochuman_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res50_ochuman_256x192.py index c95cff2f32..e91d3b9eb9 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res50_ochuman_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res50_ochuman_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/ochuman.py'] log_level = 'INFO' load_from = None resume_from = None @@ -123,19 +124,22 @@ ann_file='data/coco/annotations/person_keypoints_train2017.json', img_prefix='data/coco//train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownOCHumanDataset', ann_file=f'{data_root}/annotations/' 'ochuman_coco_format_val_range_0.00_1.00.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownOCHumanDataset', ann_file=f'{data_root}/annotations/' 'ochuman_coco_format_test_range_0.00_1.00.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res50_ochuman_384x288.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res50_ochuman_384x288.py index fcb4658a75..985a0edb18 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res50_ochuman_384x288.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/ochuman/res50_ochuman_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/ochuman.py'] log_level = 'INFO' load_from = None resume_from = None @@ -42,7 +43,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -123,19 +124,22 @@ ann_file='data/coco/annotations/person_keypoints_train2017.json', img_prefix='data/coco//train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownOCHumanDataset', ann_file=f'{data_root}/annotations/' 'ochuman_coco_format_val_range_0.00_1.00.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownOCHumanDataset', ann_file=f'{data_root}/annotations/' 'ochuman_coco_format_test_range_0.00_1.00.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/posetrack18/hrnet_w32_posetrack18_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/posetrack18/hrnet_w32_posetrack18_256x192.py index 545ccdd9c8..47284529d3 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/posetrack18/hrnet_w32_posetrack18_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/posetrack18/hrnet_w32_posetrack18_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/posetrack18.py'] log_level = 'INFO' load_from = 'https://download.openmmlab.com/mmpose/top_down/hrnet/hrnet_w32_coco_256x192-c78dce93_20200708.pth' # noqa: E501 resume_from = None @@ -155,17 +156,20 @@ ann_file=f'{data_root}/annotations/posetrack18_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownPoseTrack18Dataset', ann_file=f'{data_root}/annotations/posetrack18_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownPoseTrack18Dataset', ann_file=f'{data_root}/annotations/posetrack18_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/posetrack18/res50_posetrack18_256x192.py b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/posetrack18/res50_posetrack18_256x192.py index cff2defdb9..0fdf1ecbdb 100644 --- a/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/posetrack18/res50_posetrack18_256x192.py +++ b/configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/posetrack18/res50_posetrack18_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/posetrack18.py'] log_level = 'INFO' load_from = 'https://download.openmmlab.com/mmpose/top_down/resnet/res50_coco_256x192-ec54d7f3_20200709.pth' # noqa: E501 resume_from = None @@ -125,17 +126,20 @@ ann_file=f'{data_root}/annotations/posetrack18_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownPoseTrack18Dataset', ann_file=f'{data_root}/annotations/posetrack18_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownPoseTrack18Dataset', ann_file=f'{data_root}/annotations/posetrack18_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/3d_kpt_sview_rgb_img/pose_lift/h36m/simplebaseline3d_h36m.py b/configs/body/3d_kpt_sview_rgb_img/pose_lift/h36m/simplebaseline3d_h36m.py index 6014c1f28d..8bf60fc1b0 100644 --- a/configs/body/3d_kpt_sview_rgb_img/pose_lift/h36m/simplebaseline3d_h36m.py +++ b/configs/body/3d_kpt_sview_rgb_img/pose_lift/h36m/simplebaseline3d_h36m.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/h36m.py'] log_level = 'INFO' load_from = None resume_from = None @@ -171,17 +172,20 @@ ann_file=f'{data_root}/annotation_body3d/fps50/h36m_train.npz', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_1frame_fullconv_supervised_cpn_ft.py b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_1frame_fullconv_supervised_cpn_ft.py index 32ceefe1f5..c16df7c1fb 100644 --- a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_1frame_fullconv_supervised_cpn_ft.py +++ b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_1frame_fullconv_supervised_cpn_ft.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/h36m.py'] log_level = 'INFO' load_from = None resume_from = None @@ -141,17 +142,20 @@ ann_file=f'{data_root}/annotation_body3d/fps50/h36m_train.npz', img_prefix=f'{data_root}/images/', data_cfg=train_data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=test_data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=test_data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_243frames_fullconv_supervised.py b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_243frames_fullconv_supervised.py index b55ce3ebc7..72b011321f 100644 --- a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_243frames_fullconv_supervised.py +++ b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_243frames_fullconv_supervised.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/h36m.py'] log_level = 'INFO' load_from = None resume_from = None @@ -127,17 +128,20 @@ ann_file=f'{data_root}/annotation_body3d/fps50/h36m_train.npz', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_243frames_fullconv_supervised_cpn_ft.py b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_243frames_fullconv_supervised_cpn_ft.py index 7c07669102..c2160aa498 100644 --- a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_243frames_fullconv_supervised_cpn_ft.py +++ b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_243frames_fullconv_supervised_cpn_ft.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/h36m.py'] log_level = 'INFO' load_from = None resume_from = None @@ -141,17 +142,20 @@ ann_file=f'{data_root}/annotation_body3d/fps50/h36m_train.npz', img_prefix=f'{data_root}/images/', data_cfg=train_data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=test_data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=test_data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_27frames_fullconv_semi-supervised.py b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_27frames_fullconv_semi-supervised.py index 0a666a3914..57d95892d5 100644 --- a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_27frames_fullconv_semi-supervised.py +++ b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_27frames_fullconv_semi-supervised.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/h36m.py'] log_level = 'INFO' load_from = None resume_from = None @@ -197,23 +198,27 @@ ann_file=f'{data_root}/annotation_body3d/fps50/h36m_train.npz', img_prefix=f'{data_root}/images/', data_cfg=labeled_data_cfg, - pipeline=train_labeled_pipeline), + pipeline=train_labeled_pipeline, + dataset_info={{base.dataset_info}}), unlabeled_dataset=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_train.npz', img_prefix=f'{data_root}/images/', data_cfg=unlabeled_data_cfg, - pipeline=train_unlabeled_pipeline)), + pipeline=train_unlabeled_pipeline, + dataset_info={{base.dataset_info}})), val=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=val_data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=test_data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_27frames_fullconv_semi-supervised_cpn_ft.py b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_27frames_fullconv_semi-supervised_cpn_ft.py index c22226d43a..246929a0bd 100644 --- a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_27frames_fullconv_semi-supervised_cpn_ft.py +++ b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_27frames_fullconv_semi-supervised_cpn_ft.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/h36m.py'] log_level = 'INFO' load_from = None resume_from = None @@ -203,23 +204,27 @@ ann_file=f'{data_root}/annotation_body3d/fps50/h36m_train.npz', img_prefix=f'{data_root}/images/', data_cfg=labeled_data_cfg, - pipeline=train_labeled_pipeline), + pipeline=train_labeled_pipeline, + dataset_info={{base.dataset_info}}), unlabeled_dataset=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_train.npz', img_prefix=f'{data_root}/images/', data_cfg=unlabeled_data_cfg, - pipeline=train_unlabeled_pipeline)), + pipeline=train_unlabeled_pipeline, + dataset_info={{base.dataset_info}})), val=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=val_data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=test_data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_27frames_fullconv_supervised.py b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_27frames_fullconv_supervised.py index 6e3e8f552d..6cdda2fdb8 100644 --- a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_27frames_fullconv_supervised.py +++ b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_27frames_fullconv_supervised.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/h36m.py'] log_level = 'INFO' load_from = None resume_from = None @@ -127,17 +128,20 @@ ann_file=f'{data_root}/annotation_body3d/fps50/h36m_train.npz', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_81frames_fullconv_supervised.py b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_81frames_fullconv_supervised.py index 17d028b32c..12340e9546 100644 --- a/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_81frames_fullconv_supervised.py +++ b/configs/body/3d_kpt_sview_rgb_vid/video_pose_lift/h36m/videopose3d_h36m_81frames_fullconv_supervised.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/h36m.py'] log_level = 'INFO' load_from = None resume_from = None @@ -127,17 +128,20 @@ ann_file=f'{data_root}/annotation_body3d/fps50/h36m_train.npz', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Body3DH36MDataset', ann_file=f'{data_root}/annotation_body3d/fps50/h36m_test.npz', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=test_pipeline), + pipeline=test_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/body/3d_mesh_sview_rgb_img/hmr/mixed/res50_mixed_224x224.py b/configs/body/3d_mesh_sview_rgb_img/hmr/mixed/res50_mixed_224x224.py index 98ab35ed12..3fab96a358 100644 --- a/configs/body/3d_mesh_sview_rgb_img/hmr/mixed/res50_mixed_224x224.py +++ b/configs/body/3d_mesh_sview_rgb_img/hmr/mixed/res50_mixed_224x224.py @@ -31,7 +31,7 @@ pretrained=None, backbone=dict(type='ResNet', depth=50), mesh_head=dict( - type='MeshHMRHead', + type='HMRMeshHead', in_channels=2048, smpl_mean_params='models/smpl/smpl_mean_params.npz', ), @@ -143,7 +143,7 @@ ann_file='data/mesh_annotation_files/coco_2014_train.npz', img_prefix='data/coco', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline) ], partition=[0.35, 0.15, 0.1, 0.10, 0.10, 0.2]), adversarial_dataset=dict( diff --git a/configs/face/2d_kpt_sview_rgb_img/deeppose/wflw/res50_wflw_256x256.py b/configs/face/2d_kpt_sview_rgb_img/deeppose/wflw/res50_wflw_256x256.py index 30a33763f6..1723d8ae0d 100644 --- a/configs/face/2d_kpt_sview_rgb_img/deeppose/wflw/res50_wflw_256x256.py +++ b/configs/face/2d_kpt_sview_rgb_img/deeppose/wflw/res50_wflw_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/wflw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ backbone=dict(type='ResNet', depth=50, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -103,17 +104,20 @@ ann_file=f'{data_root}/annotations/face_landmarks_wflw_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='FaceWFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_wflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='FaceWFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_wflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/face/2d_kpt_sview_rgb_img/deeppose/wflw/res50_wflw_256x256_wingloss.py b/configs/face/2d_kpt_sview_rgb_img/deeppose/wflw/res50_wflw_256x256_wingloss.py index 0f0ee580d1..6be5101e4a 100644 --- a/configs/face/2d_kpt_sview_rgb_img/deeppose/wflw/res50_wflw_256x256_wingloss.py +++ b/configs/face/2d_kpt_sview_rgb_img/deeppose/wflw/res50_wflw_256x256_wingloss.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/wflw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ backbone=dict(type='ResNet', depth=50, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='WingLoss', use_target_weight=True)), @@ -103,17 +104,20 @@ ann_file=f'{data_root}/annotations/face_landmarks_wflw_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='FaceWFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_wflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='FaceWFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_wflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/300w/hrnetv2_w18_300w_256x256.py b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/300w/hrnetv2_w18_300w_256x256.py index 7c5eb7f76d..49a098596d 100644 --- a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/300w/hrnetv2_w18_300w_256x256.py +++ b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/300w/hrnetv2_w18_300w_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/300w.py'] log_level = 'INFO' load_from = None resume_from = None @@ -140,17 +141,20 @@ ann_file=f'{data_root}/annotations/face_landmarks_300w_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Face300WDataset', ann_file=f'{data_root}/annotations/face_landmarks_300w_valid.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Face300WDataset', ann_file=f'{data_root}/annotations/face_landmarks_300w_valid.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/300w/hrnetv2_w18_300w_256x256_dark.py b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/300w/hrnetv2_w18_300w_256x256_dark.py index d35010081e..f0c14f3b39 100644 --- a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/300w/hrnetv2_w18_300w_256x256_dark.py +++ b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/300w/hrnetv2_w18_300w_256x256_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/300w.py'] log_level = 'INFO' load_from = None resume_from = None @@ -140,17 +141,20 @@ ann_file=f'{data_root}/annotations/face_landmarks_300w_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Face300WDataset', ann_file=f'{data_root}/annotations/face_landmarks_300w_valid.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Face300WDataset', ann_file=f'{data_root}/annotations/face_landmarks_300w_valid.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/300w/res50_300w_256x256.py b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/300w/res50_300w_256x256.py index 5f46998ca9..1e5534be7b 100644 --- a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/300w/res50_300w_256x256.py +++ b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/300w/res50_300w_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/300w.py'] log_level = 'INFO' load_from = None resume_from = None @@ -106,17 +107,20 @@ ann_file=f'{data_root}/annotations/face_landmarks_300w_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Face300WDataset', ann_file=f'{data_root}/annotations/face_landmarks_300w_valid.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Face300WDataset', ann_file=f'{data_root}/annotations/face_landmarks_300w_valid.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/aflw/hrnetv2_w18_aflw_256x256.py b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/aflw/hrnetv2_w18_aflw_256x256.py index e1c88abd1d..1bb172bff5 100644 --- a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/aflw/hrnetv2_w18_aflw_256x256.py +++ b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/aflw/hrnetv2_w18_aflw_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aflw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -140,17 +141,20 @@ ann_file=f'{data_root}/annotations/face_landmarks_aflw_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='FaceAFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_aflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='FaceAFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_aflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/aflw/hrnetv2_w18_aflw_256x256_dark.py b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/aflw/hrnetv2_w18_aflw_256x256_dark.py index 0326e3da72..04c40f36f1 100644 --- a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/aflw/hrnetv2_w18_aflw_256x256_dark.py +++ b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/aflw/hrnetv2_w18_aflw_256x256_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aflw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -140,17 +141,20 @@ ann_file=f'{data_root}/annotations/face_landmarks_aflw_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='FaceAFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_aflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='FaceAFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_aflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/aflw/res50_aflw_256x256.py b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/aflw/res50_aflw_256x256.py index 0530958840..92b15f7c4d 100644 --- a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/aflw/res50_aflw_256x256.py +++ b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/aflw/res50_aflw_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/aflw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -106,17 +107,20 @@ ann_file=f'{data_root}/annotations/face_landmarks_aflw_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='FaceAFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_aflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='FaceAFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_aflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/cofw/hrnetv2_w18_cofw_256x256.py b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/cofw/hrnetv2_w18_cofw_256x256.py index 7c93465674..efa4bdf468 100644 --- a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/cofw/hrnetv2_w18_cofw_256x256.py +++ b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/cofw/hrnetv2_w18_cofw_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/cofw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -140,17 +141,20 @@ ann_file=f'{data_root}/annotations/cofw_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='FaceCOFWDataset', ann_file=f'{data_root}/annotations/cofw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='FaceCOFWDataset', ann_file=f'{data_root}/annotations/cofw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/cofw/hrnetv2_w18_cofw_256x256_dark.py b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/cofw/hrnetv2_w18_cofw_256x256_dark.py index ef61697ed2..fc86ed8538 100644 --- a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/cofw/hrnetv2_w18_cofw_256x256_dark.py +++ b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/cofw/hrnetv2_w18_cofw_256x256_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/cofw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -140,17 +141,20 @@ ann_file=f'{data_root}/annotations/cofw_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='FaceCOFWDataset', ann_file=f'{data_root}/annotations/cofw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='FaceCOFWDataset', ann_file=f'{data_root}/annotations/cofw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/cofw/res50_cofw_256x256.py b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/cofw/res50_cofw_256x256.py index 16ea04186b..3b4d6b50ba 100644 --- a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/cofw/res50_cofw_256x256.py +++ b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/cofw/res50_cofw_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/cofw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -106,17 +107,20 @@ ann_file=f'{data_root}/annotations/cofw_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='FaceCOFWDataset', ann_file=f'{data_root}/annotations/cofw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='FaceCOFWDataset', ann_file=f'{data_root}/annotations/cofw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/hrnetv2_w18_wflw_256x256.py b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/hrnetv2_w18_wflw_256x256.py index 083f72bbc7..202c64266d 100644 --- a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/hrnetv2_w18_wflw_256x256.py +++ b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/hrnetv2_w18_wflw_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/wflw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -140,17 +141,20 @@ ann_file=f'{data_root}/annotations/face_landmarks_wflw_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='FaceWFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_wflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='FaceWFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_wflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/hrnetv2_w18_wflw_256x256_dark.py b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/hrnetv2_w18_wflw_256x256_dark.py index c3504f161a..5b800531ec 100644 --- a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/hrnetv2_w18_wflw_256x256_dark.py +++ b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/hrnetv2_w18_wflw_256x256_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/wflw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -140,17 +141,20 @@ ann_file=f'{data_root}/annotations/face_landmarks_wflw_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='FaceWFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_wflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='FaceWFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_wflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/res50_wflw_256x256.py b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/res50_wflw_256x256.py index 4f7ca2645f..cb7cf374bd 100644 --- a/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/res50_wflw_256x256.py +++ b/configs/face/2d_kpt_sview_rgb_img/topdown_heatmap/wflw/res50_wflw_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/wflw.py'] log_level = 'INFO' load_from = None resume_from = None @@ -106,17 +107,20 @@ ann_file=f'{data_root}/annotations/face_landmarks_wflw_train.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='FaceWFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_wflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='FaceWFLWDataset', ann_file=f'{data_root}/annotations/face_landmarks_wflw_test.json', img_prefix=f'{data_root}/images/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res101_deepfashion_full_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res101_deepfashion_full_256x192.py index 31634423c8..b71aad0a7a 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res101_deepfashion_full_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res101_deepfashion_full_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_full.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ backbone=dict(type='ResNet', depth=101, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -117,19 +118,22 @@ img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_val.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_test.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res101_deepfashion_lower_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res101_deepfashion_lower_256x192.py index de7b6be1e0..975563099f 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res101_deepfashion_lower_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res101_deepfashion_lower_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_lower.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ backbone=dict(type='ResNet', depth=101, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -117,19 +118,22 @@ img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_val.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_test.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res101_deepfashion_upper_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res101_deepfashion_upper_256x192.py index 2fec56eeca..ba3eaac825 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res101_deepfashion_upper_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res101_deepfashion_upper_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_upper.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ backbone=dict(type='ResNet', depth=101, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -117,19 +118,22 @@ img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_val.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_test.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res152_deepfashion_full_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res152_deepfashion_full_256x192.py index c127e12c4c..3812ed282b 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res152_deepfashion_full_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res152_deepfashion_full_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_full.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ backbone=dict(type='ResNet', depth=152, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -117,19 +118,22 @@ img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_val.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_test.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res152_deepfashion_lower_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res152_deepfashion_lower_256x192.py index 78f73189b4..a9e90eba8e 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res152_deepfashion_lower_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res152_deepfashion_lower_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_lower.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ backbone=dict(type='ResNet', depth=152, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -117,19 +118,22 @@ img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_val.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_test.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res152_deepfashion_upper_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res152_deepfashion_upper_256x192.py index 069b823a13..bc64952b63 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res152_deepfashion_upper_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res152_deepfashion_upper_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_upper.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ backbone=dict(type='ResNet', depth=152, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -117,19 +118,22 @@ img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_val.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_test.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res50_deepfashion_full_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res50_deepfashion_full_256x192.py index a6511cdde3..121de540a2 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res50_deepfashion_full_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res50_deepfashion_full_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_full.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ backbone=dict(type='ResNet', depth=50, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -121,19 +122,22 @@ img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_val.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_test.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res50_deepfashion_lower_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res50_deepfashion_lower_256x192.py index d58b65fe0a..32d540b154 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res50_deepfashion_lower_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res50_deepfashion_lower_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_lower.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ backbone=dict(type='ResNet', depth=50, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -121,19 +122,22 @@ img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_val.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_test.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res50_deepfashion_upper_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res50_deepfashion_upper_256x192.py index 3f12e0415e..5b0d1b1434 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res50_deepfashion_upper_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/deeppose/deepfashion/res50_deepfashion_upper_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_upper.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ backbone=dict(type='ResNet', depth=50, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -121,19 +122,22 @@ img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_val.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_test.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_full_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_full_256x192.py index f0dc2203be..e39eeb0dd8 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_full_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_full_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_full.py'] log_level = 'INFO' load_from = None resume_from = None @@ -69,7 +70,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -151,19 +152,22 @@ img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_val.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_test.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_full_256x192_udp.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_full_256x192_udp.py index 605026a24c..96e605804c 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_full_256x192_udp.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_full_256x192_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_full.py'] log_level = 'INFO' load_from = None resume_from = None @@ -70,7 +71,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -158,19 +159,22 @@ img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_val.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_test.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_lower_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_lower_256x192.py index 556a3ec516..e125c81af6 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_lower_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_lower_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_lower.py'] log_level = 'INFO' load_from = None resume_from = None @@ -68,7 +69,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -150,19 +151,22 @@ img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_val.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_test.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_lower_256x192_udp.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_lower_256x192_udp.py index 5a205698b0..a605df63db 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_lower_256x192_udp.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_lower_256x192_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_lower.py'] log_level = 'INFO' load_from = None resume_from = None @@ -157,19 +158,22 @@ img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_val.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_test.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_upper_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_upper_256x192.py index 5effacfd7b..0e6cfb81b5 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_upper_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_upper_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_upper.py'] log_level = 'INFO' load_from = None resume_from = None @@ -69,7 +70,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -151,19 +152,22 @@ img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_val.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_test.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_upper_256x192_udp.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_upper_256x192_udp.py index 2c7e8de956..c006d3cff8 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_upper_256x192_udp.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w32_deepfashion_upper_256x192_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_upper.py'] log_level = 'INFO' load_from = None resume_from = None @@ -70,7 +71,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -158,19 +159,22 @@ img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_val.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_test.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_full_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_full_256x192.py index adbed3a8a3..753f7949fd 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_full_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_full_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_full.py'] log_level = 'INFO' load_from = None resume_from = None @@ -69,7 +70,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -151,19 +152,22 @@ img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_val.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_test.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_full_256x192_udp.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_full_256x192_udp.py index 18a238e226..eb13ac46d1 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_full_256x192_udp.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_full_256x192_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_full.py'] log_level = 'INFO' load_from = None resume_from = None @@ -70,7 +71,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -158,19 +159,22 @@ img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_val.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_test.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_lower_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_lower_256x192.py index c73306f90f..1c2885b530 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_lower_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_lower_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_lower.py'] log_level = 'INFO' load_from = None resume_from = None @@ -69,7 +70,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -151,19 +152,22 @@ img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_val.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_test.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_lower_256x192_udp.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_lower_256x192_udp.py index 42aa95f35c..ff23f99032 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_lower_256x192_udp.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_lower_256x192_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_lower.py'] log_level = 'INFO' load_from = None resume_from = None @@ -70,7 +71,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -158,19 +159,22 @@ img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_val.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_test.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_upper_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_upper_256x192.py index 34a166cf24..f053fc7137 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_upper_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_upper_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_upper.py'] log_level = 'INFO' load_from = None resume_from = None @@ -69,7 +70,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -151,19 +152,22 @@ img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_val.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_test.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_upper_256x192_udp.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_upper_256x192_udp.py index 1dda1f4055..9e04771323 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_upper_256x192_udp.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/hrnet_w48_deepfashion_upper_256x192_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_upper.py'] log_level = 'INFO' load_from = None resume_from = None @@ -70,7 +71,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -158,19 +159,22 @@ img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_val.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_test.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res101_deepfashion_full_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res101_deepfashion_full_256x192.py index 68d42d3a86..3179734e35 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res101_deepfashion_full_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res101_deepfashion_full_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_full.py'] log_level = 'INFO' load_from = None resume_from = None @@ -40,7 +41,7 @@ pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -120,19 +121,22 @@ img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_val.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_test.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res101_deepfashion_lower_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res101_deepfashion_lower_256x192.py index 883ffe6866..3ae30339bb 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res101_deepfashion_lower_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res101_deepfashion_lower_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_lower.py'] log_level = 'INFO' load_from = None resume_from = None @@ -40,7 +41,7 @@ pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -120,19 +121,22 @@ img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_val.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_test.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res101_deepfashion_upper_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res101_deepfashion_upper_256x192.py index b33d2730b4..a6c3381b60 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res101_deepfashion_upper_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res101_deepfashion_upper_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_upper.py'] log_level = 'INFO' load_from = None resume_from = None @@ -120,19 +121,22 @@ img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_val.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_test.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res152_deepfashion_full_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res152_deepfashion_full_256x192.py index 11f4596445..7e287515eb 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res152_deepfashion_full_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res152_deepfashion_full_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_full.py'] log_level = 'INFO' load_from = None resume_from = None @@ -120,19 +121,22 @@ img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_val.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_test.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res152_deepfashion_lower_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res152_deepfashion_lower_256x192.py index e45012a918..1249f979f5 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res152_deepfashion_lower_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res152_deepfashion_lower_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_lower.py'] log_level = 'INFO' load_from = None resume_from = None @@ -40,7 +41,7 @@ pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -120,19 +121,22 @@ img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_val.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_test.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res152_deepfashion_upper_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res152_deepfashion_upper_256x192.py index da939c572d..53cdc3c15b 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res152_deepfashion_upper_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res152_deepfashion_upper_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_upper.py'] log_level = 'INFO' load_from = None resume_from = None @@ -40,7 +41,7 @@ pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -120,19 +121,22 @@ img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_val.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_test.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res50_deepfashion_full_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res50_deepfashion_full_256x192.py index 3bc33ae33d..77190a4fa2 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res50_deepfashion_full_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res50_deepfashion_full_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_full.py'] log_level = 'INFO' load_from = None resume_from = None @@ -120,19 +121,22 @@ img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_val.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_full_test.json', img_prefix=f'{data_root}/img/', subset='full', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res50_deepfashion_lower_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res50_deepfashion_lower_256x192.py index d54cfc0958..e525dfa506 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res50_deepfashion_lower_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res50_deepfashion_lower_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_lower.py'] log_level = 'INFO' load_from = None resume_from = None @@ -40,7 +41,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -120,19 +121,22 @@ img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_val.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_lower_test.json', img_prefix=f'{data_root}/img/', subset='lower', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res50_deepfashion_upper_256x192.py b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res50_deepfashion_upper_256x192.py index 64398b6d94..d315aad16b 100644 --- a/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res50_deepfashion_upper_256x192.py +++ b/configs/fashion/2d_kpt_sview_rgb_img/topdown_heatmap/deepfashion/res50_deepfashion_upper_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/deepfashion_upper.py'] log_level = 'INFO' load_from = None resume_from = None @@ -120,19 +121,22 @@ img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_val.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='DeepFashionDataset', ann_file=f'{data_root}/annotations/fld_upper_test.json', img_prefix=f'{data_root}/img/', subset='upper', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/deeppose/onehand10k/res50_onehand10k_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/deeppose/onehand10k/res50_onehand10k_256x256.py index 9ad9e92e91..72268d9f91 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/deeppose/onehand10k/res50_onehand10k_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/deeppose/onehand10k/res50_onehand10k_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/onehand10k.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ backbone=dict(type='ResNet', depth=50, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -113,17 +114,20 @@ ann_file=f'{data_root}/annotations/onehand10k_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='OneHand10KDataset', ann_file=f'{data_root}/annotations/onehand10k_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='OneHand10KDataset', ann_file=f'{data_root}/annotations/onehand10k_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/deeppose/panoptic2d/res50_panoptic2d_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/deeppose/panoptic2d/res50_panoptic2d_256x256.py index 86458d4710..da3079bae2 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/deeppose/panoptic2d/res50_panoptic2d_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/deeppose/panoptic2d/res50_panoptic2d_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/panoptic_hand2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ backbone=dict(type='ResNet', depth=50, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -113,17 +114,20 @@ ann_file=f'{data_root}/annotations/panoptic_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='PanopticDataset', ann_file=f'{data_root}/annotations/panoptic_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='PanopticDataset', ann_file=f'{data_root}/annotations/panoptic_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/deeppose/rhd2d/res50_rhd2d_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/deeppose/rhd2d/res50_rhd2d_256x256.py index a95d0aae93..65807079df 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/deeppose/rhd2d/res50_rhd2d_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/deeppose/rhd2d/res50_rhd2d_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/rhd2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ backbone=dict(type='ResNet', depth=50, num_stages=4, out_indices=(3, )), neck=dict(type='GlobalAveragePooling'), keypoint_head=dict( - type='FcHead', + type='DeepposeRegressionHead', in_channels=2048, num_joints=channel_cfg['num_output_channels'], loss_keypoint=dict(type='SmoothL1Loss', use_target_weight=True)), @@ -113,17 +114,20 @@ ann_file=f'{data_root}/annotations/rhd_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/freihand2d/res50_freihand2d_224x224.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/freihand2d/res50_freihand2d_224x224.py index afb343f685..4e1bb6dbfd 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/freihand2d/res50_freihand2d_224x224.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/freihand2d/res50_freihand2d_224x224.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/freihand2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -112,17 +113,20 @@ ann_file=f'{data_root}/annotations/freihand_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='FreiHandDataset', ann_file=f'{data_root}/annotations/freihand_val.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='FreiHandDataset', ann_file=f'{data_root}/annotations/freihand_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/res50_interhand2d_all_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/res50_interhand2d_all_256x256.py index 400d6e8837..88d2a596d5 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/res50_interhand2d_all_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/res50_interhand2d_all_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/interhand2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -117,7 +118,8 @@ 'InterHand2.6M_train_joint_3d.json', img_prefix=f'{data_root}/images/train/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='InterHand2DDataset', ann_file=f'{data_root}/annotations/machine_annot/' @@ -128,7 +130,8 @@ 'InterHand2.6M_val_joint_3d.json', img_prefix=f'{data_root}/images/val/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='InterHand2DDataset', ann_file=f'{data_root}/annotations/all/' @@ -139,5 +142,6 @@ 'InterHand2.6M_test_joint_3d.json', img_prefix=f'{data_root}/images/test/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/res50_interhand2d_human_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/res50_interhand2d_human_256x256.py index 5ecb2dfbef..b3999c969e 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/res50_interhand2d_human_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/res50_interhand2d_human_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/interhand2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -117,7 +118,8 @@ 'InterHand2.6M_train_joint_3d.json', img_prefix=f'{data_root}/images/train/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='InterHand2DDataset', ann_file=f'{data_root}/annotations/machine_annot/' @@ -128,7 +130,8 @@ 'InterHand2.6M_val_joint_3d.json', img_prefix=f'{data_root}/images/val/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='InterHand2DDataset', ann_file=f'{data_root}/annotations/human_annot/' @@ -139,5 +142,6 @@ 'InterHand2.6M_test_joint_3d.json', img_prefix=f'{data_root}/images/test/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/res50_interhand2d_machine_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/res50_interhand2d_machine_256x256.py index 16df8d998b..e680b7522c 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/res50_interhand2d_machine_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/interhand2d/res50_interhand2d_machine_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/interhand2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -117,7 +118,8 @@ 'InterHand2.6M_train_joint_3d.json', img_prefix=f'{data_root}/images/train/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='InterHand2DDataset', ann_file=f'{data_root}/annotations/machine_annot/' @@ -128,7 +130,8 @@ 'InterHand2.6M_val_joint_3d.json', img_prefix=f'{data_root}/images/val/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='InterHand2DDataset', ann_file=f'{data_root}/annotations/machine_annot/' @@ -139,5 +142,6 @@ 'InterHand2.6M_test_joint_3d.json', img_prefix=f'{data_root}/images/test/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/hrnetv2_w18_onehand10k_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/hrnetv2_w18_onehand10k_256x256.py index acf08fd8a7..2298c1cbcb 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/hrnetv2_w18_onehand10k_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/hrnetv2_w18_onehand10k_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/onehand10k.py'] log_level = 'INFO' load_from = None resume_from = None @@ -146,17 +147,20 @@ ann_file=f'{data_root}/annotations/onehand10k_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='OneHand10KDataset', ann_file=f'{data_root}/annotations/onehand10k_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='OneHand10KDataset', ann_file=f'{data_root}/annotations/onehand10k_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/hrnetv2_w18_onehand10k_256x256_dark.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/hrnetv2_w18_onehand10k_256x256_dark.py index 3ffb72d7f2..66dac9c4b0 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/hrnetv2_w18_onehand10k_256x256_dark.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/hrnetv2_w18_onehand10k_256x256_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/onehand10k.py'] log_level = 'INFO' load_from = None resume_from = None @@ -76,7 +77,7 @@ multiscale_output=True), upsample=dict(mode='bilinear', align_corners=False))), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=[18, 36, 72, 144], in_index=(0, 1, 2, 3), input_transform='resize_concat', @@ -146,17 +147,20 @@ ann_file=f'{data_root}/annotations/onehand10k_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='OneHand10KDataset', ann_file=f'{data_root}/annotations/onehand10k_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='OneHand10KDataset', ann_file=f'{data_root}/annotations/onehand10k_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/hrnetv2_w18_onehand10k_256x256_udp.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/hrnetv2_w18_onehand10k_256x256_udp.py index 9b5a5b05cc..761afd7c43 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/hrnetv2_w18_onehand10k_256x256_udp.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/hrnetv2_w18_onehand10k_256x256_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/onehand10k.py'] log_level = 'INFO' load_from = None resume_from = None @@ -77,7 +78,7 @@ multiscale_output=True), upsample=dict(mode='bilinear', align_corners=False))), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=[18, 36, 72, 144], in_index=(0, 1, 2, 3), input_transform='resize_concat', @@ -153,17 +154,20 @@ ann_file=f'{data_root}/annotations/onehand10k_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='OneHand10KDataset', ann_file=f'{data_root}/annotations/onehand10k_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='OneHand10KDataset', ann_file=f'{data_root}/annotations/onehand10k_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/mobilenetv2_onehand10k_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/mobilenetv2_onehand10k_256x256.py index b7d64eb849..7d9444fd10 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/mobilenetv2_onehand10k_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/mobilenetv2_onehand10k_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/onehand10k.py'] log_level = 'INFO' load_from = None resume_from = None @@ -113,17 +114,20 @@ ann_file=f'{data_root}/annotations/onehand10k_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='OneHand10KDataset', ann_file=f'{data_root}/annotations/onehand10k_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='OneHand10KDataset', ann_file=f'{data_root}/annotations/onehand10k_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/res50_onehand10k_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/res50_onehand10k_256x256.py index bb06d2f7fe..214a3dc85a 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/res50_onehand10k_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/onehand10k/res50_onehand10k_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/onehand10k.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -112,17 +113,20 @@ ann_file=f'{data_root}/annotations/onehand10k_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='OneHand10KDataset', ann_file=f'{data_root}/annotations/onehand10k_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='OneHand10KDataset', ann_file=f'{data_root}/annotations/onehand10k_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/hrnetv2_w18_panoptic2d_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/hrnetv2_w18_panoptic2d_256x256.py index c4b3e29c97..39d30a5d06 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/hrnetv2_w18_panoptic2d_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/hrnetv2_w18_panoptic2d_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/panoptic_hand2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -146,17 +147,20 @@ ann_file=f'{data_root}/annotations/panoptic_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='PanopticDataset', ann_file=f'{data_root}/annotations/panoptic_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='PanopticDataset', ann_file=f'{data_root}/annotations/panoptic_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/hrnetv2_w18_panoptic2d_256x256_dark.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/hrnetv2_w18_panoptic2d_256x256_dark.py index bec68dbe48..fdfe451413 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/hrnetv2_w18_panoptic2d_256x256_dark.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/hrnetv2_w18_panoptic2d_256x256_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/panoptic_hand2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -146,17 +147,20 @@ ann_file=f'{data_root}/annotations/panoptic_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='PanopticDataset', ann_file=f'{data_root}/annotations/panoptic_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='PanopticDataset', ann_file=f'{data_root}/annotations/panoptic_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/hrnetv2_w18_panoptic2d_256x256_udp.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/hrnetv2_w18_panoptic2d_256x256_udp.py index a0c7ef0f38..0fecf01328 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/hrnetv2_w18_panoptic2d_256x256_udp.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/hrnetv2_w18_panoptic2d_256x256_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/panoptic_hand2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -153,17 +154,20 @@ ann_file=f'{data_root}/annotations/panoptic_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='PanopticDataset', ann_file=f'{data_root}/annotations/panoptic_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='PanopticDataset', ann_file=f'{data_root}/annotations/panoptic_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/mobilenetv2_panoptic2d_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/mobilenetv2_panoptic2d_256x256.py index e85c4d0790..31eca71ed4 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/mobilenetv2_panoptic2d_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/mobilenetv2_panoptic2d_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/panoptic_hand2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -112,17 +113,20 @@ ann_file=f'{data_root}/annotations/panoptic_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='PanopticDataset', ann_file=f'{data_root}/annotations/panoptic_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='PanopticDataset', ann_file=f'{data_root}/annotations/panoptic_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/res50_panoptic2d_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/res50_panoptic2d_256x256.py index 5206dc1ec7..072a361ec8 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/res50_panoptic2d_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/panoptic2d/res50_panoptic2d_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/panoptic_hand2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -112,17 +113,20 @@ ann_file=f'{data_root}/annotations/panoptic_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='PanopticDataset', ann_file=f'{data_root}/annotations/panoptic_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='PanopticDataset', ann_file=f'{data_root}/annotations/panoptic_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/hrnetv2_w18_rhd2d_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/hrnetv2_w18_rhd2d_256x256.py index 6f5957d7ac..f101ee934e 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/hrnetv2_w18_rhd2d_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/hrnetv2_w18_rhd2d_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/rhd2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -76,7 +77,7 @@ multiscale_output=True), upsample=dict(mode='bilinear', align_corners=False))), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=[18, 36, 72, 144], in_index=(0, 1, 2, 3), input_transform='resize_concat', @@ -146,17 +147,20 @@ ann_file=f'{data_root}/annotations/rhd_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/hrnetv2_w18_rhd2d_256x256_dark.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/hrnetv2_w18_rhd2d_256x256_dark.py index 98e388eedd..7c24e46327 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/hrnetv2_w18_rhd2d_256x256_dark.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/hrnetv2_w18_rhd2d_256x256_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/rhd2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -76,7 +77,7 @@ multiscale_output=True), upsample=dict(mode='bilinear', align_corners=False))), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=[18, 36, 72, 144], in_index=(0, 1, 2, 3), input_transform='resize_concat', @@ -146,17 +147,20 @@ ann_file=f'{data_root}/annotations/rhd_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/hrnetv2_w18_rhd2d_256x256_udp.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/hrnetv2_w18_rhd2d_256x256_udp.py index 836014a393..e939db750e 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/hrnetv2_w18_rhd2d_256x256_udp.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/hrnetv2_w18_rhd2d_256x256_udp.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/rhd2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -77,7 +78,7 @@ multiscale_output=True), upsample=dict(mode='bilinear', align_corners=False))), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=[18, 36, 72, 144], in_index=(0, 1, 2, 3), input_transform='resize_concat', @@ -153,17 +154,20 @@ ann_file=f'{data_root}/annotations/rhd_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/mobilenetv2_rhd2d_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/mobilenetv2_rhd2d_256x256.py index 236cbb8180..3ee36f1fd2 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/mobilenetv2_rhd2d_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/mobilenetv2_rhd2d_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/rhd2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ pretrained='mmcls://mobilenet_v2', backbone=dict(type='MobileNetV2', widen_factor=1., out_indices=(7, )), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=1280, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -112,17 +113,20 @@ ann_file=f'{data_root}/annotations/rhd_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/res50_rhd2d_224x224.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/res50_rhd2d_224x224.py index e663cbc678..c70e77b6ad 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/res50_rhd2d_224x224.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/res50_rhd2d_224x224.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/rhd2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -112,17 +113,20 @@ ann_file=f'{data_root}/annotations/rhd_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/res50_rhd2d_256x256.py b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/res50_rhd2d_256x256.py index 2fe58cd4cd..596caf8042 100644 --- a/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/res50_rhd2d_256x256.py +++ b/configs/hand/2d_kpt_sview_rgb_img/topdown_heatmap/rhd2d/res50_rhd2d_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/rhd2d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -47,7 +48,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -112,17 +113,20 @@ ann_file=f'{data_root}/annotations/rhd_train.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='Rhd2DDataset', ann_file=f'{data_root}/annotations/rhd_test.json', img_prefix=f'{data_root}/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/hand/3d_kpt_sview_rgb_img/internet/interhand3d/res50_interhand3d_all_256x256.py b/configs/hand/3d_kpt_sview_rgb_img/internet/interhand3d/res50_interhand3d_all_256x256.py index 6334f7345b..3ebcea8a97 100644 --- a/configs/hand/3d_kpt_sview_rgb_img/internet/interhand3d/res50_interhand3d_all_256x256.py +++ b/configs/hand/3d_kpt_sview_rgb_img/internet/interhand3d/res50_interhand3d_all_256x256.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/interhand3d.py'] log_level = 'INFO' load_from = None resume_from = None @@ -149,7 +150,8 @@ data_cfg=data_cfg, use_gt_root_depth=True, rootnet_result_file=None, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='InterHand3DDataset', ann_file=f'{data_root}/annotations/machine_annot/' @@ -162,7 +164,8 @@ data_cfg=data_cfg, use_gt_root_depth=True, rootnet_result_file=None, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='InterHand3DDataset', ann_file=f'{data_root}/annotations/all/' @@ -175,5 +178,6 @@ data_cfg=data_cfg, use_gt_root_depth=True, rootnet_result_file=None, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w32_coco_wholebody_512x512.py b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w32_coco_wholebody_512x512.py index db89340076..112fb11b7f 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w32_coco_wholebody_512x512.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w32_coco_wholebody_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=32, num_joints=133, tag_per_joint=True, @@ -181,17 +182,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w32_coco_wholebody_640x640.py b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w32_coco_wholebody_640x640.py index 4483e6eb81..e220b11a02 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w32_coco_wholebody_640x640.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w32_coco_wholebody_640x640.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=32, num_joints=133, tag_per_joint=True, @@ -181,17 +182,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w48_coco_wholebody_512x512.py b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w48_coco_wholebody_512x512.py index b05b0ad812..9cc927c5f3 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w48_coco_wholebody_512x512.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w48_coco_wholebody_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w48-8ef0771d.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=48, num_joints=133, tag_per_joint=True, @@ -181,17 +182,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w48_coco_wholebody_640x640.py b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w48_coco_wholebody_640x640.py index 38a6938ab3..2458febc74 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w48_coco_wholebody_640x640.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/higherhrnet_w48_coco_wholebody_640x640.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w48-8ef0771d.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='BottomUpHigherResolutionHead', + type='AEHigherResolutionHead', in_channels=48, num_joints=133, tag_per_joint=True, @@ -181,17 +182,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w32_coco_wholebody_512x512.py b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w32_coco_wholebody_512x512.py index 06ff2edf81..062e8e83de 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w32_coco_wholebody_512x512.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w32_coco_wholebody_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=32, num_joints=133, num_deconv_layers=0, @@ -177,17 +178,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w32_coco_wholebody_640x640.py b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w32_coco_wholebody_640x640.py index 595ecb0052..acd7549e5f 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w32_coco_wholebody_640x640.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w32_coco_wholebody_640x640.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w32-36af842e.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=32, num_joints=133, num_deconv_layers=0, @@ -177,17 +178,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w48_coco_wholebody_512x512.py b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w48_coco_wholebody_512x512.py index a9a567b48f..e7a53d9eaf 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w48_coco_wholebody_512x512.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w48_coco_wholebody_512x512.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w48-8ef0771d.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=48, num_joints=133, num_deconv_layers=0, @@ -177,17 +178,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w48_coco_wholebody_640x640.py b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w48_coco_wholebody_640x640.py index c320d625b2..c12a96bb9d 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w48_coco_wholebody_640x640.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/associative_embedding/coco-wholebody/hrnet_w48_coco_wholebody_640x640.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -48,7 +49,7 @@ # model settings model = dict( - type='BottomUp', + type='AssociativeEmbedding', pretrained='https://download.openmmlab.com/mmpose/' 'pretrain_models/hrnet_w48-8ef0771d.pth', backbone=dict( @@ -81,7 +82,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='BottomUpSimpleHead', + type='AESimpleHead', in_channels=48, num_joints=133, num_deconv_layers=0, @@ -177,17 +178,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='BottomUpCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_256x192.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_256x192.py index 2e77501591..f05978ac48 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_256x192.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -69,7 +70,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -152,17 +153,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_256x192_dark.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_256x192_dark.py index 0ef39c4d48..ef0f2147ab 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_256x192_dark.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_256x192_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -69,7 +70,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -152,17 +153,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_384x288.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_384x288.py index 93d8d88e7a..f5e1ccee67 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_384x288.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -69,7 +70,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -152,17 +153,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_384x288_dark.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_384x288_dark.py index f7620c6165..d50dec859a 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_384x288_dark.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w32_coco_wholebody_384x288_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -69,7 +70,7 @@ num_channels=(32, 64, 128, 256))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=32, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -152,17 +153,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_256x192.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_256x192.py index 92f32e00bc..88fbc24b39 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_256x192.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -69,7 +70,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -152,17 +153,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_256x192_dark.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_256x192_dark.py index 5d357956b9..4fe14aef1d 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_256x192_dark.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_256x192_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -69,7 +70,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -152,17 +153,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_384x288.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_384x288.py index cdb3225abf..6c7235f388 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_384x288.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -70,7 +71,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -153,17 +154,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_384x288_dark.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_384x288_dark.py index a27c1a569d..e6a6877cea 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_384x288_dark.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_384x288_dark.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -69,7 +70,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -152,17 +153,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_384x288_dark_plus.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_384x288_dark_plus.py index 8766f29e9b..4e0d8bd9c4 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_384x288_dark_plus.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/hrnet_w48_coco_wholebody_384x288_dark_plus.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -69,7 +70,7 @@ num_channels=(48, 96, 192, 384))), ), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=48, out_channels=channel_cfg['num_output_channels'], num_deconv_layers=0, @@ -152,17 +153,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res101_coco_wholebody_256x192.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res101_coco_wholebody_256x192.py index 9fd1695c44..4eb16df078 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res101_coco_wholebody_256x192.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res101_coco_wholebody_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -40,7 +41,7 @@ pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -121,17 +122,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res101_coco_wholebody_384x288.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res101_coco_wholebody_384x288.py index a89bee7c02..3a45e2b62d 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res101_coco_wholebody_384x288.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res101_coco_wholebody_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ pretrained='torchvision://resnet101', backbone=dict(type='ResNet', depth=101), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -122,17 +123,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res152_coco_wholebody_256x192.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res152_coco_wholebody_256x192.py index 5a4212991d..4974b2c852 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res152_coco_wholebody_256x192.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res152_coco_wholebody_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -40,7 +41,7 @@ pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -121,17 +122,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res152_coco_wholebody_384x288.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res152_coco_wholebody_384x288.py index dd80a403f7..b4fd1db812 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res152_coco_wholebody_384x288.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res152_coco_wholebody_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -41,7 +42,7 @@ pretrained='torchvision://resnet152', backbone=dict(type='ResNet', depth=152), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -122,17 +123,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res50_coco_wholebody_256x192.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res50_coco_wholebody_256x192.py index 8eec68010c..3da0504fb3 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res50_coco_wholebody_256x192.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res50_coco_wholebody_256x192.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -40,7 +41,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -121,17 +122,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), ) diff --git a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res50_coco_wholebody_384x288.py b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res50_coco_wholebody_384x288.py index 4e768f3cb5..ff6d4f1c22 100644 --- a/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res50_coco_wholebody_384x288.py +++ b/configs/wholebody/2d_kpt_sview_rgb_img/topdown_heatmap/coco-wholebody/res50_coco_wholebody_384x288.py @@ -1,3 +1,4 @@ +_base_ = base = ['../../../../_base_/datasets/coco_wholebody.py'] log_level = 'INFO' load_from = None resume_from = None @@ -40,7 +41,7 @@ pretrained='torchvision://resnet50', backbone=dict(type='ResNet', depth=50), keypoint_head=dict( - type='TopDownSimpleHead', + type='TopdownHeatmapSimpleHead', in_channels=2048, out_channels=channel_cfg['num_output_channels'], loss_keypoint=dict(type='JointsMSELoss', use_target_weight=True)), @@ -121,17 +122,20 @@ ann_file=f'{data_root}/annotations/coco_wholebody_train_v1.0.json', img_prefix=f'{data_root}/train2017/', data_cfg=data_cfg, - pipeline=train_pipeline), + pipeline=train_pipeline, + dataset_info={{base.dataset_info}}), val=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), test=dict( type='TopDownCocoWholeBodyDataset', ann_file=f'{data_root}/annotations/coco_wholebody_val_v1.0.json', img_prefix=f'{data_root}/val2017/', data_cfg=data_cfg, - pipeline=val_pipeline), + pipeline=val_pipeline, + dataset_info={{base.dataset_info}}), )