From 55551f1fe77b589abf44c3c030e38628311b4f86 Mon Sep 17 00:00:00 2001 From: franz101 Date: Mon, 13 Nov 2023 20:13:56 -0800 Subject: [PATCH] fix sci-kit depencey --- deepfloyd_if/pipelines/inpainting.py | 6 +++--- requirements.txt | 1 + 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/deepfloyd_if/pipelines/inpainting.py b/deepfloyd_if/pipelines/inpainting.py index d0a4c78..c491763 100644 --- a/deepfloyd_if/pipelines/inpainting.py +++ b/deepfloyd_if/pipelines/inpainting.py @@ -58,7 +58,7 @@ def inpainting( if_I_kwargs['support_noise'] = low_res - inpainting_mask_I = img_as_bool(resize(inpainting_mask[0].cpu(), (3, image_h, image_w))) + inpainting_mask_I = img_as_bool(resize(inpainting_mask[0].cpu(), (3, image_h, image_w), anti_aliasing=True)) inpainting_mask_I = torch.from_numpy(inpainting_mask_I).unsqueeze(0).to(if_I.device) if_I_kwargs['inpainting_mask'] = inpainting_mask_I @@ -81,7 +81,7 @@ def inpainting( if_II_kwargs['support_noise'] = mid_res if 'inpainting_mask' not in if_II_kwargs: - inpainting_mask_II = img_as_bool(resize(inpainting_mask[0].cpu(), (3, image_h, image_w))) + inpainting_mask_II = img_as_bool(resize(inpainting_mask[0].cpu(), (3, image_h, image_w), anti_aliasing=True)) inpainting_mask_II = torch.from_numpy(inpainting_mask_II).unsqueeze(0).to(if_II.device) if_II_kwargs['inpainting_mask'] = inpainting_mask_II @@ -110,7 +110,7 @@ def inpainting( if_III_kwargs['support_noise'] = high_res if 'inpainting_mask' not in if_III_kwargs: - inpainting_mask_III = img_as_bool(resize(inpainting_mask[0].cpu(), (3, image_h, image_w))) + inpainting_mask_III = img_as_bool(resize(inpainting_mask[0].cpu(), (3, image_h, image_w), anti_aliasing=True)) inpainting_mask_III = torch.from_numpy(inpainting_mask_III).unsqueeze(0).to(if_III.device) if_III_kwargs['inpainting_mask'] = inpainting_mask_III diff --git a/requirements.txt b/requirements.txt index 8fd0cbb..fccea61 100644 --- a/requirements.txt +++ b/requirements.txt @@ -13,3 +13,4 @@ tokenizers~=0.13.2 sentencepiece~=0.1.97 ftfy~=6.1.1 beautifulsoup4~=4.11.1 +scikit-image>=0.22.0 \ No newline at end of file