diff --git a/dezero/datasets.py b/dezero/datasets.py index 677d15e..77b8374 100644 --- a/dezero/datasets.py +++ b/dezero/datasets.py @@ -232,7 +232,7 @@ def _load_label(self, filename, data_type='train'): def labels(label_type='fine'): coarse_labels = dict(enumerate(['aquatic mammals','fish','flowers','food containers','fruit and vegetables','household electrical device','household furniture','insects','large carnivores','large man-made outdoor things','large natural outdoor scenes','large omnivores and herbivores','medium-sized mammals','non-insect invertebrates','people','reptiles','small mammals','trees','vehicles 1','vehicles 2'])) fine_labels = [] - return fine_labels if label_type is 'fine' else coarse_labels + return fine_labels if label_type == 'fine' else coarse_labels