update keypoint

open-mmlab · Jun 9, 2021 · a1c3e06 · a1c3e06
1 parent 48aa0bd
commit a1c3e06
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mmpose/datasets/datasets/_base_/kpt_3d_sview_kpt_2d_dataset.py b/mmpose/datasets/datasets/_base_/kpt_3d_sview_kpt_2d_dataset.py
@@ -65,6 +65,7 @@ def __init__(self,
 
         self.load_config(self.data_cfg)
 
+        self.ann_info['num_joints'] = data_cfg['num_joints']
         assert self.ann_info['num_joints'] == dataset_info.keypoint_num
         self.ann_info['flip_pairs'] = dataset_info.flip_pairs
         self.ann_info['upper_body_ids'] = dataset_info.upper_body_ids