NOBODY commited on
Commit
1ee5e0e
·
1 Parent(s): ea42584

fix(data): dataset error when running on test-dev (#1137)

Browse files
Files changed (1) hide show
  1. yolox/data/datasets/coco.py +3 -2
yolox/data/datasets/coco.py CHANGED
@@ -27,8 +27,9 @@ def remove_useless_info(coco):
27
  img.pop("coco_url", None)
28
  img.pop("date_captured", None)
29
  img.pop("flickr_url", None)
30
- for anno in coco.dataset["annotations"]:
31
- anno.pop("segmentation", None)
 
32
 
33
 
34
  class COCODataset(Dataset):
 
27
  img.pop("coco_url", None)
28
  img.pop("date_captured", None)
29
  img.pop("flickr_url", None)
30
+ if "annotations" in coco.dataset:
31
+ for anno in coco.dataset["annotations"]:
32
+ anno.pop("segmentation", None)
33
 
34
 
35
  class COCODataset(Dataset):