Fix dtype, remove unnecessary conversion

microsoft · Jul 9, 2022 · eb1f061 · eb1f061
1 parent 025001e
commit eb1f061
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchgeo/datasets/reforestree.py b/torchgeo/datasets/reforestree.py
@@ -152,7 +152,7 @@ def _load_image(self, path: str) -> Tensor:
             the image
         """
         with Image.open(path) as img:
-            array: "np.typing.NDArray[np.int_]" = np.array(img.convert("RGB"))
+            array: "np.typing.NDArray[np.uint8]" = np.array(img)
             tensor = torch.from_numpy(array)
             # Convert from HxWxC to CxHxW
             tensor = tensor.permute((2, 0, 1))