Skip to content

Commit 614c046

Browse files
committed
aligned split naming with paper
1 parent 2c00160 commit 614c046

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tensorflow_datasets/image_classification/wake_vision/wake_vision_dataset_builder.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,8 @@
2727
)
2828
]
2929
,
30-
'train_image_metadata': 'https://dataverse.harvard.edu/api/access/datafile/9844933?format=original',
31-
'train_bbox_metadata': 'https://dataverse.harvard.edu/api/access/datafile/9844934?format=original',
30+
'train_large_metadata': 'https://dataverse.harvard.edu/api/access/datafile/9844933?format=original',
31+
'train_quality_metadata': 'https://dataverse.harvard.edu/api/access/datafile/9844934?format=original',
3232
'validation_metadata': 'https://dataverse.harvard.edu/api/access/datafile/10069808?format=original',
3333
'test_metadata': 'https://dataverse.harvard.edu/api/access/datafile/10069809?format=original',
3434
}
@@ -90,8 +90,8 @@ def _split_generators(self, dl_manager: tfds.download.DownloadManager):
9090
paths = dl_manager.download_and_extract(_URLS)
9191

9292
return {
93-
'train_image': self._generate_examples(paths['train_images'], paths['train_image_metadata']),
94-
'train_bbox': self._generate_examples(paths['train_images'], paths['train_bbox_metadata']),
93+
'train_large': self._generate_examples(paths['train_images'], paths['train_large_metadata']),
94+
'train_quality': self._generate_examples(paths['train_images'], paths['train_quality_metadata']),
9595
'validation' : self._generate_examples(paths['validation_images'], paths['validation_metadata']),
9696
'test' : self._generate_examples(paths['test_images'], paths['test_metadata']),
9797
}

0 commit comments

Comments
 (0)