Fix coordinate swap bug in IDTReeS (#683)

* Fix coordinate swap * Test coverage
microsoft · Sep 3, 2022 · ec72bfe · ec72bfe
1 parent 1044078
commit ec72bfe
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/torchgeo/datasets/idtrees.py b/torchgeo/datasets/idtrees.py
@@ -289,10 +289,10 @@ def _load_boxes(self, path: str) -> Tensor:
         with rasterio.open(path) as f:
             for geom in geoms:
                 coords = [f.index(x, y) for x, y in geom]
-                xmin = min(coord[0] for coord in coords)
-                xmax = max(coord[0] for coord in coords)
-                ymin = min(coord[1] for coord in coords)
-                ymax = max(coord[1] for coord in coords)
+                xmin = min(coord[1] for coord in coords)
+                xmax = max(coord[1] for coord in coords)
+                ymin = min(coord[0] for coord in coords)
+                ymax = max(coord[0] for coord in coords)
                 boxes.append([xmin, ymin, xmax, ymax])
 
         tensor = torch.tensor(boxes)