fix mask from list

2024-04-29 11:55:16 +02:00
parent 9f42fafd31
commit 1a1b4bf5c4
1 changed files with 2 additions and 1 deletions
@@ -793,7 +793,8 @@ class MaskFromList:
            values = [values]

        values = torch.tensor(values).float()
-        values = (values - values.min()) / values.max()
+        values = torch.clamp(values, 0.0, 1.0)
+        #values = (values - values.min()) / values.max()

        return (values.unsqueeze(1).unsqueeze(2).repeat(1, width, height), )