From ed443624baf4784cb8f4b7c8718c7698eef3fbf7 Mon Sep 17 00:00:00 2001 From: cubiq Date: Wed, 14 Aug 2024 16:56:59 +0200 Subject: [PATCH] clamp values to 0-1 in the random transform node --- image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/image.py b/image.py index 44a037d..69afa31 100644 --- a/image.py +++ b/image.py @@ -743,7 +743,7 @@ class ImageRandomTransform: ]) out.append(tramsforms(i.unsqueeze(0))) - out = torch.cat(out, dim=0).permute([0, 2, 3, 1]) + out = torch.cat(out, dim=0).permute([0, 2, 3, 1]).clamp(0, 1) return (out,)