fix listing directories bug x2

microsoft · Apr 12, 2023 · b47c62a · b47c62a
1 parent d033fa6
commit b47c62a
Showing 1 changed file with 1 addition and 1 deletion.
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 = [f for f in os.listdir(root) if os.path.isdir(f)]
+        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]