ignore kid annotations

open-mmlab · Jun 27, 2022 · f210de9 · f210de9
1 parent a6b3b70
commit f210de9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mmhuman3d/data/data_converters/agora.py b/mmhuman3d/data/data_converters/agora.py
@@ -212,6 +212,10 @@ def convert_by_mode(self, dataset_path: str, out_path: str,
                     occlusion = df.iloc[idx]['occlusion'][pidx]
                     ethnicity = df.iloc[idx]['ethnicity'][pidx]
 
+                    # skip kid since they use different model
+                    if kid:
+                        continue
+
                     # obtain keypoints
                     keypoints2d = df.iloc[idx]['gt_joints_2d'][pidx]
                     if self.res == (1280, 720):