Adjust naming scheme for SSL4EO dataset (#1362)

* outline downstream dataset

* revert commit

* fix naming scheme

tests/conf/ssl4eo_l_byol_2.yaml

@@ -8,8 +8,8 @@ module:
 
 datamodule:
   _target_: torchgeo.datamodules.SSL4EOLDataModule
-  root: "tests/data/ssl4eo/l/tm_sr"
-  split: "tm_sr"
+  root: "tests/data/ssl4eo/l/etm_sr"
+  split: "etm_sr"
   seasons: 2
   batch_size: 2
   num_workers: 0

tests/data/ssl4eo/l/data.py

@@ -22,19 +22,19 @@ FILENAME_HIERARCHY = Union[dict[str, "FILENAME_HIERARCHY"], list[str]]
 filenames: FILENAME_HIERARCHY = {
     "tm_toa": {
         "0000002": {
-            "LE07_172034_20010526": ["all_bands.tif"],
-            "LE07_172034_20020310": ["all_bands.tif"],
-            "LE07_172034_20020902": ["all_bands.tif"],
-            "LE07_172034_20021121": ["all_bands.tif"],
+            "LT05_172034_20010526": ["all_bands.tif"],
+            "LT05_172034_20020310": ["all_bands.tif"],
+            "LT05_172034_20020902": ["all_bands.tif"],
+            "LT05_172034_20021121": ["all_bands.tif"],
         },
         "0000005": {
-            "LE07_223084_20010413": ["all_bands.tif"],
-            "LE07_223084_20011225": ["all_bands.tif"],
-            "LE07_223084_20020619": ["all_bands.tif"],
-            "LE07_223084_20020923": ["all_bands.tif"],
+            "LT05_223084_20010413": ["all_bands.tif"],
+            "LT05_223084_20011225": ["all_bands.tif"],
+            "LT05_223084_20020619": ["all_bands.tif"],
+            "LT5_223084_20020923": ["all_bands.tif"],
         },
     },
-    "tm_sr": {
+    "etm_sr": {
         "0000002": {
             "LE07_172034_20010526": ["all_bands.tif"],
             "LE07_172034_20020310": ["all_bands.tif"],
@@ -92,7 +92,7 @@ filenames: FILENAME_HIERARCHY = {
     },
 }
 
-num_bands = {"tm_toa": 7, "tm_sr": 6, "etm_toa": 9, "oli_tirs_toa": 11, "oli_sr": 7}
+num_bands = {"tm_toa": 7, "etm_sr": 6, "etm_toa": 9, "oli_tirs_toa": 11, "oli_sr": 7}
 
 
 def create_file(path: str) -> None:

torchgeo/datasets/ssl4eo.py

@@ -86,7 +86,7 @@ class SSL4EOL(NonGeoDataset):
 
     metadata: dict[str, _Metadata] = {
         "tm_toa": {"num_bands": 7, "rgb_bands": [2, 1, 0]},
-        "tm_sr": {"num_bands": 6, "rgb_bands": [2, 1, 0]},
+        "etm_sr": {"num_bands": 6, "rgb_bands": [2, 1, 0]},
         "etm_toa": {"num_bands": 9, "rgb_bands": [2, 1, 0]},
         "oli_tirs_toa": {"num_bands": 11, "rgb_bands": [3, 2, 1]},
         "oli_sr": {"num_bands": 7, "rgb_bands": [3, 2, 1]},
@@ -103,7 +103,7 @@ class SSL4EOL(NonGeoDataset):
 
         Args:
             root: root directory where dataset can be found
-            split: one of ['tm_toa', 'tm_sr', 'etm_toa', 'oli_tirs_toa', 'oli_sr']
+            split: one of ['tm_toa', 'etm_sr', 'etm_toa', 'oli_tirs_toa', 'oli_sr']
             seasons: number of seasonal patches to sample per location, 1--4
             transforms: a function/transform that takes input sample and its target as
                 entry and returns a transformed version