From 060308c3e58d943b580a4f194e35f2d9566408e6 Mon Sep 17 00:00:00 2001 From: Wei-Chen-hub <1259566226@qq.com> Date: Tue, 31 Oct 2023 17:46:42 +0800 Subject: [PATCH] fix --- mmhuman3d/data/data_converters/agora.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmhuman3d/data/data_converters/agora.py b/mmhuman3d/data/data_converters/agora.py index b10940d5..c7950084 100644 --- a/mmhuman3d/data/data_converters/agora.py +++ b/mmhuman3d/data/data_converters/agora.py @@ -124,7 +124,7 @@ def convert_by_mode(self, dataset_path: str, out_path: str, ]: bboxs_[bbox_name] = [] meta_ = {} - for meta_key in ['principal_point', 'focal_length', 'gender']: + for meta_key in ['principal_point', 'focal_length', 'gender', 'annot_valid']: meta_[meta_key] = [] image_path_ = [] @@ -172,7 +172,6 @@ def convert_by_mode(self, dataset_path: str, out_path: str, # for aid, anno_info in enumerate(tqdm(anno_param, desc=f'Processing Agora {mode}')): image_id = anno_info['image_id'] - pdb.set_trace() # anno_info_bf = anno_param_bf[aid] # print(anno_info['gender']) @@ -232,6 +231,7 @@ def convert_by_mode(self, dataset_path: str, out_path: str, meta_['gender'].append(anno_info['gender']) meta_['principal_point'].append(principal_point) meta_['focal_length'].append(focal_length) + meta_['annot_valid'].append(anno_info['is_valid']) # print(smplx_param['betas'].shape)