Unverified Commit bdc75e75 authored by Will Berman's avatar Will Berman Committed by GitHub
Browse files

[IF super res] correctly normalize PIL input (#3536)

* [IF super res] correctl normalize PIL input

* 175 -> 127.5
parent 1d1f648c
...@@ -759,7 +759,7 @@ class IFImg2ImgSuperResolutionPipeline(DiffusionPipeline): ...@@ -759,7 +759,7 @@ class IFImg2ImgSuperResolutionPipeline(DiffusionPipeline):
image = [image] image = [image]
if isinstance(image[0], PIL.Image.Image): if isinstance(image[0], PIL.Image.Image):
image = [np.array(i).astype(np.float32) / 255.0 for i in image] image = [np.array(i).astype(np.float32) / 127.5 - 1.0 for i in image]
image = np.stack(image, axis=0) # to np image = np.stack(image, axis=0) # to np
image = torch.from_numpy(image.transpose(0, 3, 1, 2)) image = torch.from_numpy(image.transpose(0, 3, 1, 2))
......
...@@ -795,7 +795,7 @@ class IFInpaintingSuperResolutionPipeline(DiffusionPipeline): ...@@ -795,7 +795,7 @@ class IFInpaintingSuperResolutionPipeline(DiffusionPipeline):
image = [image] image = [image]
if isinstance(image[0], PIL.Image.Image): if isinstance(image[0], PIL.Image.Image):
image = [np.array(i).astype(np.float32) / 255.0 for i in image] image = [np.array(i).astype(np.float32) / 127.5 - 1.0 for i in image]
image = np.stack(image, axis=0) # to np image = np.stack(image, axis=0) # to np
image = torch.from_numpy(image.transpose(0, 3, 1, 2)) image = torch.from_numpy(image.transpose(0, 3, 1, 2))
......
...@@ -664,7 +664,7 @@ class IFSuperResolutionPipeline(DiffusionPipeline): ...@@ -664,7 +664,7 @@ class IFSuperResolutionPipeline(DiffusionPipeline):
image = [image] image = [image]
if isinstance(image[0], PIL.Image.Image): if isinstance(image[0], PIL.Image.Image):
image = [np.array(i).astype(np.float32) / 255.0 for i in image] image = [np.array(i).astype(np.float32) / 127.5 - 1.0 for i in image]
image = np.stack(image, axis=0) # to np image = np.stack(image, axis=0) # to np
image = torch.from_numpy(image.transpose(0, 3, 1, 2)) image = torch.from_numpy(image.transpose(0, 3, 1, 2))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment