clamp values to 0-1 in the random transform node

2024-08-14 16:56:59 +02:00
parent 87ec9ae07b
commit ed443624ba
1 changed files with 1 additions and 1 deletions
@@ -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,)