Format to float (#1656)

microsoft · Oct 13, 2023 · 82b5921 · 82b5921
1 parent 27be503
commit 82b5921
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchgeo/datasets/oscd.py b/torchgeo/datasets/oscd.py
@@ -204,7 +204,7 @@ def _load_image(self, paths: Sequence[str]) -> Tensor:
             with Image.open(path) as img:
                 images.append(np.array(img))
         array: "np.typing.NDArray[np.int_]" = np.stack(images, axis=0).astype(np.int_)
-        tensor = torch.from_numpy(array)
+        tensor = torch.from_numpy(array).float()
         return tensor
 
     def _load_target(self, path: str) -> Tensor: