From 730b2e81ed6ee1a37f73b84c6393ef03fda46855 Mon Sep 17 00:00:00 2001 From: vmalefioudakis Date: Wed, 23 Oct 2024 22:35:55 +0300 Subject: [PATCH] update inference_loop.py --- GANDLF/data/inference_dataloader_histopath.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GANDLF/data/inference_dataloader_histopath.py b/GANDLF/data/inference_dataloader_histopath.py index d3285deea..1f34c9a74 100644 --- a/GANDLF/data/inference_dataloader_histopath.py +++ b/GANDLF/data/inference_dataloader_histopath.py @@ -136,7 +136,7 @@ def __getitem__(self, idx): self._selected_level, (self._patch_size[0], self._patch_size[1]), # as_array=True, openslide-python doesn't return a ndarray, return an image - ) + ).convert('RGB') patch = np.asarray(patch) # convert the image to ndarray # this is to ensure that channels come at the beginning