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

fix(data): dataset error when running on test-dev
Megvii-BaseDetection · Feb 22, 2022 · 8a9003d · 8a9003d
1 parent d98da01
commit 8a9003d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/yolox/data/datasets/coco.py b/yolox/data/datasets/coco.py
@@ -27,8 +27,9 @@ def remove_useless_info(coco):
             img.pop("coco_url", None)
             img.pop("date_captured", None)
             img.pop("flickr_url", None)
-        for anno in coco.dataset["annotations"]:
-            anno.pop("segmentation", None)
+        if "annotations" in coco.dataset:
+            for anno in coco.dataset["annotations"]:
+                anno.pop("segmentation", None)
 
 
 class COCODataset(Dataset):