Some images in the SeCo dataset are not square (#658)

microsoft · Jul 7, 2022 · 111921c · 111921c
1 parent 49d6114
commit 111921c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/torchgeo/datasets/seco.py b/torchgeo/datasets/seco.py
@@ -172,8 +172,7 @@ def _load_patch(self, scene_name: str, patch_name: str) -> Tensor:
             with rasterio.open(fn) as f:
                 band_data = f.read(1)
                 height, width = band_data.shape
-                assert height == width
-                size = height
+                size = min(height, width)
                 if size < 264:
                     # TODO: PIL resize is much slower than cv2, we should check to see
                     # what could be sped up throughout later. There is also a potential