From e9c048c3ab32729509ae9ed554cffc0de12cb4ea Mon Sep 17 00:00:00 2001 From: Walter Hugo Lopez Pinaya Date: Sat, 18 Mar 2023 19:11:22 +0000 Subject: [PATCH] Clip image_data values before casting to uint8 Signed-off-by: Walter Hugo Lopez Pinaya --- .../cxr_image_synthesis_latent_diffusion_model/scripts/saver.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model-zoo/models/cxr_image_synthesis_latent_diffusion_model/scripts/saver.py b/model-zoo/models/cxr_image_synthesis_latent_diffusion_model/scripts/saver.py index b945ceec..05e88722 100644 --- a/model-zoo/models/cxr_image_synthesis_latent_diffusion_model/scripts/saver.py +++ b/model-zoo/models/cxr_image_synthesis_latent_diffusion_model/scripts/saver.py @@ -11,7 +11,7 @@ def __init__(self, output_dir: str) -> None: self.output_dir = output_dir def save(self, image_data: torch.Tensor, file_name: str) -> None: - image_data = image_data.cpu().numpy() + image_data = np.clip(image_data.cpu().numpy(), 0, 1) image_data = (image_data * 255).astype(np.uint8) im = Image.fromarray(image_data[0, 0]) im.save(self.output_dir + "/" + file_name + ".jpg")