Skip to content

GitLab

Unverified Commit ed0d4689 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

fix Flowers102 dataset labels (#5775)

parent bc9924fe
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/datasets/flowers102.py
View file @ ed0d4689
...@@ -65,7 +65,7 @@ class Flowers102(VisionDataset): ...@@ -65,7 +65,7 @@ class Flowers102(VisionDataset):
image_ids = set_ids[self._splits_map[self._split]].tolist() image_ids = set_ids[self._splits_map[self._split]].tolist()
labels = loadmat(self._base_folder / self._file_dict["label"][0], squeeze_me=True) 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._labels = []
self._image_files = [] self._image_files = []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment