From 1161a5c6daa6223e38e7cb8d78a79c59912636f7 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Sun, 20 Aug 2023 17:39:48 +0900 Subject: [PATCH] fix debug_dataset for controlnet dataset --- library/train_util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/library/train_util.py b/library/train_util.py index ff1b4a33..c5a605d9 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -1976,7 +1976,7 @@ def debug_dataset(train_dataset, show_input_ids=False): if "conditioning_images" in example: cond_img = example["conditioning_images"][j] print(f"conditioning image size: {cond_img.size()}") - cond_img = (cond_img.numpy() * 255.0).astype(np.uint8) + cond_img = ((cond_img.numpy() + 1.0) * 127.5).astype(np.uint8) cond_img = np.transpose(cond_img, (1, 2, 0)) cond_img = cond_img[:, :, ::-1] if os.name == "nt":