diff --git a/GANDLF/data/inference_dataloader_histopath.py b/GANDLF/data/inference_dataloader_histopath.py index b93bd8cda..602a939ed 100644 --- a/GANDLF/data/inference_dataloader_histopath.py +++ b/GANDLF/data/inference_dataloader_histopath.py @@ -63,9 +63,8 @@ def _basic_preprocessing(self): mask = get_tissue_mask(np.asarray( self._os_image.read_region( (0, 0), self._mask_level, (mask_xdim, mask_ydim) - )) + ).convert('RGB')) ) - if self._selected_level != self._mask_level: mask = resize(mask, (height, width)) mask = (mask > 0).astype(np.ubyte)