tfds-nightly 4.9.9.dev202509050044__py3-none-any.whl → 4.9.9.dev202509070044__py3-none-any.whl

tensorflow_datasets/datasets/lvis/checksums.tsv

@@ -4,4 +4,5 @@ http://images.cocodataset.org/zips/val2017.zip	815585330	4f7e2ccb2866ec5041993c9
 https://dl.fbaipublicfiles.com/LVIS/lvis_v1_image_info_test_dev.json.zip	384629	bbe4880cef768f3c14ea903d31dcce9db24916cf9e9f1f76d1a022120f47254b	lvis_v1_image_info_test_dev.json.zip
 https://dl.fbaipublicfiles.com/LVIS/lvis_v1_train.json.zip	350264821	334a4caa374030a7817cf050364525e910f7960f9b6968cef47cffbf3893f8ba	lvis_v1_train.json.zip
 https://dl.fbaipublicfiles.com/LVIS/lvis_v1_val.json.zip	64026968	5cae9a3c79aadb667550c2b5dcf7f4d86e059a41ec91ef690225b667e28e9ba5	lvis_v1_val.json.zip
+https://huggingface.co/GLIPModel/GLIP/resolve/main/lvis_v1_minival_inserted_image_name.json	35463626	02301f6ccd89d1ee3d35112cb57d000c3396f34e4073066c90b2c1fbf47b55ce	lvis_v1_minival_inserted_image_name.json
 https://nyu.box.com/shared/static/2yk9x8az9pnlsy2v8gd95yncwn2q7vj6.zip	1587847	13b6a7121a1d9716490b1e2778c4866d640e9059b01de5ca1277fdda7f70967d	lvis_v1_minival.zip

tensorflow_datasets/datasets/lvis/lvis_dataset_builder.py

@@ -25,7 +25,7 @@ from etils import epath
 import numpy as np
 import tensorflow_datasets.public_api as tfds
 
-_URLS = {
+_EXTRACT_URLS = {
     'train_annotation': (
         'https://dl.fbaipublicfiles.com/LVIS/lvis_v1_train.json.zip'
     ),
@@ -34,11 +34,16 @@ _URLS = {
         'https://dl.fbaipublicfiles.com/LVIS/lvis_v1_val.json.zip'
     ),
     'validation_images': 'http://images.cocodataset.org/zips/val2017.zip',
-    'test_annotation': 'https://dl.fbaipublicfiles.com/LVIS/lvis_v1_image_info_test_dev.json.zip',
+    'test_annotation': (
+        'https://dl.fbaipublicfiles.com/LVIS/lvis_v1_image_info_test_dev.json.zip'
+    ),
     'test_images': 'http://images.cocodataset.org/zips/test2017.zip',
-    # Minival from https://github.com/ashkamath/mdetr/blob/main/.github/lvis.md:
+}
+_URLS = {
+    # Minival from
+    #   https://gitlab.com/YuanHuan_Pro/YOLO-World/-/blob/master/docs/data.md:
     'minival_annotation': (
-        'https://nyu.box.com/shared/static/2yk9x8az9pnlsy2v8gd95yncwn2q7vj6.zip'
+        'https://huggingface.co/GLIPModel/GLIP/resolve/main/lvis_v1_minival_inserted_image_name.json'
     ),
 }
 
@@ -71,13 +76,14 @@ _NUM_CLASSES = 1203
 class Builder(tfds.core.GeneratorBasedBuilder):
   """DatasetBuilder for lvis dataset."""
 
-  VERSION = tfds.core.Version('1.3.0')
+  VERSION = tfds.core.Version('1.4.0')
   RELEASE_NOTES = {
       '1.1.0': (
           'Added fields `neg_category_ids` and `not_exhaustive_category_ids`.'
       ),
       '1.2.0': 'Added class names.',
       '1.3.0': 'Added minival split.',
+      '1.4.0': 'Added segmentation masks to the minival split.',
   }
 
   def _info(self) -> tfds.core.DatasetInfo:
@@ -116,7 +122,10 @@ class Builder(tfds.core.GeneratorBasedBuilder):
 
   def _split_generators(self, dl_manager: tfds.download.DownloadManager):
     """Returns SplitGenerators."""
-    paths = dl_manager.download_and_extract(_URLS)
+    paths = {
+        **dl_manager.download_and_extract(_EXTRACT_URLS),
+        **dl_manager.download(_URLS),
+    }
     image_dirs = [
         paths['train_images'] / 'train2017',
         paths['validation_images'] / 'val2017',
@@ -135,7 +144,7 @@ class Builder(tfds.core.GeneratorBasedBuilder):
         ),
         'minival': self._generate_examples(
             image_dirs,
-            paths['minival_annotation'] / 'lvis_v1_minival.json',
+            paths['minival_annotation'],
         ),
     }
 
@@ -169,7 +178,7 @@ class Builder(tfds.core.GeneratorBasedBuilder):
             'bbox': _build_bbox(image_info, *inst['bbox']),
             'label': inst['category_id'] - 1,
             'segmentation': _build_segmentation_mask(
-                image_info, inst.get('segmentation', [])  # No segs in minival.
+                image_info, inst['segmentation']
             ),
         })
       return image_info['id'], example

tensorflow_datasets/datasets/lvis/lvis_dataset_builder_test.py

@@ -37,7 +37,7 @@ class LvisTest(tfds.testing.DatasetBuilderTestCase):
       'validation_images': '',
       'test_annotation': '',
       'test_images': '',
-      'minival_annotation': '',
+      'minival_annotation': 'lvis_v1_minival_inserted_image_name.json',
   }
 
 

{tfds_nightly-4.9.9.dev202509050044.dist-info → tfds_nightly-4.9.9.dev202509070044.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tfds-nightly
-Version: 4.9.9.dev202509050044
+Version: 4.9.9.dev202509070044
 Summary: tensorflow/datasets is a library of datasets ready to use with TensorFlow.
 Home-page: https://github.com/tensorflow/datasets
 Download-URL: https://github.com/tensorflow/datasets/tags

{tfds_nightly-4.9.9.dev202509050044.dist-info → tfds_nightly-4.9.9.dev202509070044.dist-info}/RECORD

@@ -884,10 +884,10 @@ tensorflow_datasets/datasets/lvis/CITATIONS.bib,sha256=48fTd93Fe8UaVahU01fVefBsl
 tensorflow_datasets/datasets/lvis/README.md,sha256=Ydfa6grVWv5dw7mVX09Yg88xG5qMmrqXl5r-3cqCf28,60
 tensorflow_datasets/datasets/lvis/TAGS.txt,sha256=rHkvxLMpnZvL9FJlJRfs-fwA_gawmkWIEXq9h_uHY2o,107
 tensorflow_datasets/datasets/lvis/__init__.py,sha256=eFqnTjU7s5iubj6XcKoU8lZUSHecOdnebZFm1vTkjbA,612
-tensorflow_datasets/datasets/lvis/checksums.tsv,sha256=b0tIEJY77MVUPQb8YQe-LY9n5UeAaq4GfFtcG54w6ag,1066
+tensorflow_datasets/datasets/lvis/checksums.tsv,sha256=7e9huxxgNCNfo0QxWJY_h5OBTS3RFBPeQOEWZCm_dpw,1273
 tensorflow_datasets/datasets/lvis/classes.txt,sha256=7AKQ2Gp9drf3BXya8H6iD7eIiz4agfXDIAqgcQYoG40,11977
-tensorflow_datasets/datasets/lvis/lvis_dataset_builder.py,sha256=_FTPSWWHXNYBJwdi97ww8Z3NPqrlMcMzH6RtFeSuR3o,8282
-tensorflow_datasets/datasets/lvis/lvis_dataset_builder_test.py,sha256=mGrPx8MRYYx3EX-rp-04a0uODPmP2di78JRugML7W2Y,1326
+tensorflow_datasets/datasets/lvis/lvis_dataset_builder.py,sha256=CnSJTCqfdcUV8lHjRBS2_yMM6nH-bj0q1sjpLekUfNU,8430
+tensorflow_datasets/datasets/lvis/lvis_dataset_builder_test.py,sha256=c46IQ-9_Oa0wTZ2QC4pMaxxiJ6zeC94l0omO8uKdST4,1366
 tensorflow_datasets/datasets/malaria/CITATIONS.bib,sha256=px7oYLvEExRjvZcUiPzFBokiX3KylAAgrDD0-x49eqo,452
 tensorflow_datasets/datasets/malaria/README.md,sha256=HJ6zPanQIpbgMtVs2cwOt4TPgtI46RB4a86Fo8nCRlw,175
 tensorflow_datasets/datasets/malaria/TAGS.txt,sha256=VGiea2bi3F9sXp1_nt3az-dKBMyNvrzhZ5qu1GgbGOY,52
@@ -2471,10 +2471,10 @@ tensorflow_datasets/vision_language/wit/wit_test.py,sha256=PXS8DMNW-MDrT2p5oy4Ic
 tensorflow_datasets/vision_language/wit_kaggle/__init__.py,sha256=vGwSGeM8WE4Q-l0-eEE1sBojmk6YT0l1OO60AWa4Q40,719
 tensorflow_datasets/vision_language/wit_kaggle/wit_kaggle.py,sha256=q-vX_FBzIwsFxL4sY9vuyQ3UQD2PLM4yhUR4U6l-qao,16903
 tensorflow_datasets/vision_language/wit_kaggle/wit_kaggle_test.py,sha256=ZymHT1NkmD-pUnh3BmM3_g30c5afsWYnmqDD9dVyDSA,1778
-tfds_nightly-4.9.9.dev202509050044.dist-info/licenses/AUTHORS,sha256=nvBG4WwfgjuOu1oZkuQKw9kg7X6rve679ObS-YDDmXg,309
-tfds_nightly-4.9.9.dev202509050044.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-tfds_nightly-4.9.9.dev202509050044.dist-info/METADATA,sha256=Dx2fO9sNwwPxgdiK0CMzcCny7IWhRw5b3Wb9eWFkbsI,11291
-tfds_nightly-4.9.9.dev202509050044.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-tfds_nightly-4.9.9.dev202509050044.dist-info/entry_points.txt,sha256=eHEL7nF5y1uCY2FgkuYIdE062epJXlAQTSdq89px4p4,73
-tfds_nightly-4.9.9.dev202509050044.dist-info/top_level.txt,sha256=bAevmk9209s_oxVZVlN6hSDIVS423qrMQvmcWSvW4do,20
-tfds_nightly-4.9.9.dev202509050044.dist-info/RECORD,,
+tfds_nightly-4.9.9.dev202509070044.dist-info/licenses/AUTHORS,sha256=nvBG4WwfgjuOu1oZkuQKw9kg7X6rve679ObS-YDDmXg,309
+tfds_nightly-4.9.9.dev202509070044.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+tfds_nightly-4.9.9.dev202509070044.dist-info/METADATA,sha256=yppZ1a0sRZKLho0e22QDdA1OkgSylSeX6kfQOFJd_3o,11291
+tfds_nightly-4.9.9.dev202509070044.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tfds_nightly-4.9.9.dev202509070044.dist-info/entry_points.txt,sha256=eHEL7nF5y1uCY2FgkuYIdE062epJXlAQTSdq89px4p4,73
+tfds_nightly-4.9.9.dev202509070044.dist-info/top_level.txt,sha256=bAevmk9209s_oxVZVlN6hSDIVS423qrMQvmcWSvW4do,20
+tfds_nightly-4.9.9.dev202509070044.dist-info/RECORD,,