|
27 | 27 | )
|
28 | 28 | ]
|
29 | 29 | ,
|
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', |
32 | 32 | 'validation_metadata': 'https://dataverse.harvard.edu/api/access/datafile/10069808?format=original',
|
33 | 33 | 'test_metadata': 'https://dataverse.harvard.edu/api/access/datafile/10069809?format=original',
|
34 | 34 | }
|
@@ -90,8 +90,8 @@ def _split_generators(self, dl_manager: tfds.download.DownloadManager):
|
90 | 90 | paths = dl_manager.download_and_extract(_URLS)
|
91 | 91 |
|
92 | 92 | 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']), |
95 | 95 | 'validation' : self._generate_examples(paths['validation_images'], paths['validation_metadata']),
|
96 | 96 | 'test' : self._generate_examples(paths['test_images'], paths['test_metadata']),
|
97 | 97 | }
|
|
0 commit comments