From bc9608148740585b971f6b82620e225320e4710f Mon Sep 17 00:00:00 2001 From: lvmin Date: Thu, 10 Aug 2023 18:54:55 -0700 Subject: [PATCH] i --- modules/core.py | 2 +- modules/cv2win32.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/modules/core.py b/modules/core.py index b3801308..99762469 100644 --- a/modules/core.py +++ b/modules/core.py @@ -79,7 +79,7 @@ def get_previewer(device, latent_format): with torch.no_grad(): x_sample = taesd.decoder(torch.nn.functional.avg_pool2d(x0, kernel_size=(2, 2))).detach() * 255.0 x_sample = einops.rearrange(x_sample, 'b c h w -> b h w c') - x_sample = x_sample.cpu().numpy()[..., ::-1].copy().clip(0, 255).astype(np.uint8) + x_sample = x_sample.cpu().numpy().clip(0, 255).astype(np.uint8) for i, s in enumerate(x_sample): show_preview(f'OpenCV Diffusion Preview {i}', s, title=f'Preview Image {i} [{step}/{total_steps}]') diff --git a/modules/cv2win32.py b/modules/cv2win32.py index bc85abc4..a235fd78 100644 --- a/modules/cv2win32.py +++ b/modules/cv2win32.py @@ -4,7 +4,7 @@ import cv2 def show_preview(flag, img, title=None): if title is None: title = flag - cv2.imshow(flag, img) + cv2.imshow(flag, img[:, :, ::-1].copy()) cv2.setWindowTitle(flag, title) cv2.setWindowProperty(flag, cv2.WND_PROP_TOPMOST, 1) cv2.waitKey(1)