diff --git a/mmhuman3d/data/data_converters/h36m.py b/mmhuman3d/data/data_converters/h36m.py index c15cb19e..333b450c 100644 --- a/mmhuman3d/data/data_converters/h36m.py +++ b/mmhuman3d/data/data_converters/h36m.py @@ -234,6 +234,15 @@ def convert_by_mode(self, dataset_path: str, out_path: str, for user_i in tqdm(user_list, desc='user id'): user_name = f'S{user_i}' # path with GT bounding boxes + # sometimes the path of GT bbox may be ground_truth_bb + if os.path.exists( + os.path.join(dataset_path, user_name, 'MySegmentsMat', + 'ground_truth_bb')): + os.rename( + os.path.join(dataset_path, user_name, 'MySegmentsMat', + 'ground_truth_bb'), + os.path.join(dataset_path, user_name, 'MySegmentsMat', + 'ground_truth_bs')) bbox_path = os.path.join(dataset_path, user_name, 'MySegmentsMat', 'ground_truth_bs') # path with GT 2D pose