From ba820e350cd3f2e403312014f0cf83937910a369 Mon Sep 17 00:00:00 2001
From: Konrad Heidler <k.heidler@tum.de>
Date: Mon, 11 Apr 2022 22:10:49 +0200
Subject: [PATCH] Cast uint16 tensors to int32, leave float32 tensors as they
 are (#502)

---
 torchgeo/datasets/sen12ms.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/torchgeo/datasets/sen12ms.py b/torchgeo/datasets/sen12ms.py
index 2c21357d519..44e47af86ad 100644
--- a/torchgeo/datasets/sen12ms.py
+++ b/torchgeo/datasets/sen12ms.py
@@ -266,7 +266,9 @@ def _load_raster(self, filename: str, source: str) -> Tensor:
                 "{}_{}_{}_{}_{}".format(*parts),
             )
         ) as f:
-            array = f.read().astype(np.int32)
+            array = f.read()
+            if array.dtype == np.uint16:
+                array = array.astype(np.int32)
             tensor = torch.from_numpy(array)
             return tensor