Skip to content

Commit

Permalink
Fix Visual Domain Decathlon dataset labels.
Browse files Browse the repository at this point in the history
By mistake, label=0 was being replaced with -1.

PiperOrigin-RevId: 263512606
  • Loading branch information
jpuigcerver authored and copybara-github committed Aug 15, 2019
1 parent 1010b12 commit bd34837
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions tensorflow_datasets/image/visual_domain_decathlon.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ class VisualDomainDecathlonConfig(tfds.core.BuilderConfig):
def __init__(self, num_classes, **kwargs):
self.num_classes = num_classes
if 'version' not in kwargs:
kwargs['version'] = tfds.core.Version('1.0.0')
kwargs['version'] = tfds.core.Version('1.1.0')
super(VisualDomainDecathlonConfig, self).__init__(**kwargs)


Expand Down Expand Up @@ -157,7 +157,6 @@ def _generate_examples(self, images_archive, annotations_path, split):
path=images_archive, method=tfds.download.ExtractMethod.TAR_STREAM):
if image_fname in filename_to_label:
label = filename_to_label[image_fname]
label = label or -1 # Note: No labels for test images.
example = {
'name': image_fname,
'image': image_fobj,
Expand Down

0 comments on commit bd34837

Please sign in to comment.