download openimages by running it (#5396)

2024-07-11 23:06:13 +00:00 · 2024-07-11 23:06:13 +00:00 · a02b38c0ac
parent 0421f5d83e
commit a02b38c0ac
1 changed files with 4 additions and 0 deletions
--- a/extra/datasets/openimages.py
+++ b/extra/datasets/openimages.py
@ -175,3 +175,7 @@ def iterate(coco, bs=8):
      annotations = coco.loadAnns(coco.getAnnIds(img_id))
      targets.append(prepare_target(annotations, img_id, original_size))
    yield np.array(X), targets
+
+if __name__ == "__main__":
+  openimages("validation")
+  openimages("train")