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