We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bc9924f commit ed0d468Copy full SHA for ed0d468
torchvision/datasets/flowers102.py
@@ -65,7 +65,7 @@ def __init__(
65
image_ids = set_ids[self._splits_map[self._split]].tolist()
66
67
labels = loadmat(self._base_folder / self._file_dict["label"][0], squeeze_me=True)
68
- image_id_to_label = dict(enumerate(labels["labels"].tolist(), 1))
+ image_id_to_label = dict(enumerate((labels["labels"] - 1).tolist(), 1))
69
70
self._labels = []
71
self._image_files = []
0 commit comments