IDTReeS - Clip boxes outside of image bounds

torchgeo/datasets/idtrees.py

@@ -296,6 +296,7 @@ def _load_boxes(self, path: str) -> Tensor:
                 boxes.append([xmin, ymin, xmax, ymax])
 
         tensor = torch.tensor(boxes)
+        tensor = torch.clamp(tensor, min=0, max=self.image_size[0])
         return tensor
 
     def _load_target(self, path: str) -> Tensor: