convert self.dataset.bounds to self.index.bounds to incorporate roi l…

…imits
sede-open · Sep 20, 2024 · 83f96b8 · 83f96b8
1 parent 29c3dc8
commit 83f96b8
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/torchgeo/samplers/single.py b/torchgeo/samplers/single.py
@@ -143,8 +143,8 @@ def __iter__(self) -> Iterator[BoundingBox]:
 
             bounds = hit.bounds
             if self.dataset.return_as_ts:
-                mint = self.dataset.bounds.mint
-                maxt = self.dataset.bounds.maxt
+                mint = self.index.bounds.mint
+                maxt = self.index.bounds.maxt
                 bounds[-2] = mint
                 bounds[-1] = maxt
 
@@ -242,8 +242,8 @@ def __iter__(self) -> Iterator[BoundingBox]:
             rows, cols = tile_to_chips(bounds, self.size, self.stride)
 
             if self.dataset.return_as_ts:
-                mint = self.dataset.bounds.mint
-                maxt = self.dataset.bounds.maxt
+                mint = self.index.bounds.mint
+                maxt = self.index.bounds.maxt
             else:
                 mint = bounds.mint
                 maxt = bounds.maxt
@@ -317,8 +317,8 @@ def __iter__(self) -> Iterator[BoundingBox]:
             bounding_box = self.hits[idx].bounds
 
             if self.dataset.return_as_ts:
-                mint = self.dataset.bounds.mint
-                maxt = self.dataset.bounds.maxt
+                mint = self.index.bounds.mint
+                maxt = self.index.bounds.maxt
             else:
                 mint = bounding_box.mint
                 maxt = bounding_box.maxt