Skip to content

Commit fad24b8

Browse files
committed
refactor: Removed name mapping
1 parent cbcd204 commit fad24b8

File tree

1 file changed

+0
-10
lines changed

1 file changed

+0
-10
lines changed

torchvision/datasets/eurosat.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -22,15 +22,6 @@ class EuroSAT(ImageFolder):
2222
url = "https://madm.dfki.de/files/sentinel/EuroSAT.zip"
2323
md5 = "c8fa014336c82ac7804f0398fcb19387"
2424

25-
_class_map = {
26-
"AnnualCrop": "Annual Crop",
27-
"HerbaceousVegetation": "Herbaceous Vegetation",
28-
"Industrial": "Industrial Buildings",
29-
"PermanentCrop": "Permanent Crop",
30-
"Residential": "Residential Buildings",
31-
"SeaLake": "Sea & Lake",
32-
}
33-
3425
def __init__(
3526
self,
3627
root: str,
@@ -48,7 +39,6 @@ def __init__(
4839
raise RuntimeError("Dataset not found. You can use download=True to download it")
4940

5041
super().__init__(self._data_folder, **kwargs)
51-
self.classes = [self._class_map.get(cls, cls) for cls in self.classes]
5242
self.root = os.path.expanduser(root)
5343

5444
def __len__(self) -> int:

0 commit comments

Comments
 (0)