SeCo newer version bug fix (microsoft#1235)

yichiac · Apr 14, 2023 · 7d3e16b · 7d3e16b
1 parent 265a236
commit 7d3e16b
Show file tree

Hide file tree

Showing 16 changed files with 2 additions and 2 deletions.
diff --git a/tests/data/seco/seasonal_contrast_100k/000000/20190803T154559_20190803T154611_T18QVG.tif b/tests/data/seco/seasonal_contrast_100k/000000/20190803T154559_20190803T154611_T18QVG.tif
diff --git a/tests/data/seco/seasonal_contrast_100k/000000/20191027T154551_20191027T154553_T18QVG.tif b/tests/data/seco/seasonal_contrast_100k/000000/20191027T154551_20191027T154553_T18QVG.tif
diff --git a/tests/data/seco/seasonal_contrast_100k/000000/20191027T154551_20191027T154553_T18QVG/B1.tif b/tests/data/seco/seasonal_contrast_100k/000000/20191027T154551_20191027T154553_T18QVG/B1.tif
diff --git a/tests/data/seco/seasonal_contrast_100k/000000/20200120T154549_20200120T154543_T18QVH.tif b/tests/data/seco/seasonal_contrast_100k/000000/20200120T154549_20200120T154543_T18QVH.tif
diff --git a/tests/data/seco/seasonal_contrast_100k/000000/20200120T154549_20200120T154543_T18QVH/B1.tif b/tests/data/seco/seasonal_contrast_100k/000000/20200120T154549_20200120T154543_T18QVH/B1.tif
diff --git a/tests/data/seco/seasonal_contrast_100k/000000/20200414T154551_20200414T154549_T18QVG.tif b/tests/data/seco/seasonal_contrast_100k/000000/20200414T154551_20200414T154549_T18QVG.tif
diff --git a/tests/data/seco/seasonal_contrast_100k/000000/20200414T154551_20200414T154549_T18QVG/B1.tif b/tests/data/seco/seasonal_contrast_100k/000000/20200414T154551_20200414T154549_T18QVG/B1.tif
diff --git a/tests/data/seco/seasonal_contrast_100k/000000/20200623T154601_20200623T154555_T18QVH.tif b/tests/data/seco/seasonal_contrast_100k/000000/20200623T154601_20200623T154555_T18QVH.tif
diff --git a/tests/data/seco/seasonal_contrast_100k/000001/20190214T071009_20190214T072118_T39RUJ.tif b/tests/data/seco/seasonal_contrast_100k/000001/20190214T071009_20190214T072118_T39RUJ.tif
diff --git a/tests/data/seco/seasonal_contrast_100k/000001/20190515T070629_20190515T071534_T39RUJ.tif b/tests/data/seco/seasonal_contrast_100k/000001/20190515T070629_20190515T071534_T39RUJ.tif
diff --git a/tests/data/seco/seasonal_contrast_100k/000001/20190803T070629_20190803T071729_T39RUJ.tif b/tests/data/seco/seasonal_contrast_100k/000001/20190803T070629_20190803T071729_T39RUJ.tif
diff --git a/tests/data/seco/seasonal_contrast_100k/000001/20191030T072031_20191030T072025_T39RUJ.tif b/tests/data/seco/seasonal_contrast_100k/000001/20191030T072031_20191030T072025_T39RUJ.tif
diff --git a/tests/data/seco/seasonal_contrast_100k/000001/20200108T072301_20200108T072255_T39RUJ.tif b/tests/data/seco/seasonal_contrast_100k/000001/20200108T072301_20200108T072255_T39RUJ.tif
diff --git a/tests/data/seco/seco_100k.zip b/tests/data/seco/seco_100k.zip
diff --git a/tests/datasets/test_seco.py b/tests/datasets/test_seco.py
@@ -42,7 +42,7 @@ def dataset(
         monkeypatch.setitem(
             SeasonalContrastS2.metadata["100k"],
             "md5",
-            "4d3e6e4afed7e581b7de1bfa2f7c29da",
+            "6f527567f066562af2c78093114599f9",
         )
         monkeypatch.setitem(
             SeasonalContrastS2.metadata["1m"],

diff --git a/torchgeo/datasets/seco.py b/torchgeo/datasets/seco.py
@@ -126,7 +126,7 @@ def __getitem__(self, index: int) -> Dict[str, Tensor]:
         root = os.path.join(
             self.root, self.metadata[self.version]["directory"], f"{index:06}"
         )
-        subdirs = os.listdir(root)
+        subdirs = [f for f in os.listdir(root) if os.path.isdir(os.path.join(root, f))]
         subdirs = random.sample(subdirs, self.seasons)
 
         images = [self._load_patch(root, subdir) for subdir in subdirs]