fix: remove useless keys in dataset

KYjoy · Jan 9, 2023 · d61f54a · d61f54a
1 parent 26d41ce
commit d61f54a
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/damo/dataset/datasets/coco.py b/damo/dataset/datasets/coco.py
@@ -47,9 +47,6 @@ def __getitem__(self, inp):
         classes = torch.tensor(classes)
         target.add_field('labels', classes)
 
-        if anno and 'keypoints' in anno[0]:
-            keypoints = [obj['keypoints'] for obj in anno]
-            target.add_field('keypoints', keypoints)
 
         target = target.clip_to_image(remove_empty=True)