NOBODY
commited on
Commit
·
1ee5e0e
1
Parent(s):
ea42584
fix(data): dataset error when running on test-dev (#1137)
Browse files
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 |
-
|
31 |
-
anno.
|
|
|
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):
|