From 6e8a0f208265a6e58a3e3d500fdc9f41a6060080 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Tue, 20 Sep 2022 19:29:17 -0700 Subject: [PATCH] fix auto-conversion of images to mode in dataset --- denoising_diffusion_pytorch/denoising_diffusion_pytorch.py | 4 ++-- setup.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py index 6e98474..31a44e4 100644 --- a/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py +++ b/denoising_diffusion_pytorch/denoising_diffusion_pytorch.py @@ -56,7 +56,7 @@ def num_to_groups(num, divisor): arr.append(remainder) return arr -def convert_image_to(img_type, image): +def convert_image_to_fn(img_type, image): if image.mode != img_type: return image.convert(img_type) return image @@ -704,7 +704,7 @@ class Dataset(Dataset): self.image_size = image_size self.paths = [p for ext in exts for p in Path(f'{folder}').glob(f'**/*.{ext}')] - maybe_convert_fn = partial(convert_image_to, convert_image_to) if exists(convert_image_to) else nn.Identity() + maybe_convert_fn = partial(convert_image_to_fn, convert_image_to) if exists(convert_image_to) else nn.Identity() self.transform = T.Compose([ T.Lambda(maybe_convert_fn), diff --git a/setup.py b/setup.py index 0ecd2f6..6bc32d8 100644 --- a/setup.py +++ b/setup.py @@ -3,7 +3,7 @@ from setuptools import setup, find_packages setup( name = 'denoising-diffusion-pytorch', packages = find_packages(), - version = '0.27.10', + version = '0.27.11', license='MIT', description = 'Denoising Diffusion Probabilistic Models - Pytorch', author = 'Phil Wang',