fix Flowers102 dataset labels (#5775)

ed0d4689 · Philip Meier · GitHub · bc9924fe · ed0d4689
Unverified Commit ed0d4689 authored Apr 07, 2022 by Philip Meier Committed by GitHub Apr 07, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/datasets/flowers102.py torchvision/datasets/flowers102.py +1 -1

No files found.
--- a/torchvision/datasets/flowers102.py
+++ b/torchvision/datasets/flowers102.py
@@ -65,7 +65,7 @@ class Flowers102(VisionDataset):
        image_ids = set_ids[self._splits_map[self._split]].tolist()

        labels = loadmat(self._base_folder / self._file_dict["label"][0], squeeze_me=True)
-        image_id_to_label = dict(enumerate(labels["labels"].tolist(), 1))
+        image_id_to_label = dict(enumerate((labels["labels"] - 1).tolist(), 1))

        self._labels = []
        self._image_files = []