much stricter mask

cloneofsimo · cloneofsimo · commit cb4b6532950a · 2023-02-04T15:31:48.000Z
diff --git a/lora_diffusion/dataset.py b/lora_diffusion/dataset.py
@@ -267,7 +267,7 @@ def __getitem__(self, index):
                     Image.open(self.mask_path[index % self.num_instance_images])
                 )
                 * 0.5
-                + 1
+                + 0.5
             )
 
         if self.h_flip and random.random() > 0.5:

Original file line number	Diff line number	Diff line change
`@@ -267,7 +267,7 @@ def __getitem__(self, index):`
`267`	`267`	`Image.open(self.mask_path[index % self.num_instance_images])`
`268`	`268`	`)`
`269`	`269`	`* 0.5`
`270`		`- + 1`
	`270`	`+ + 0.5`
`271`	`271`	`)`
`272`	`272`
`273`	`273`	`if self.h_flip and random.random() > 0.5:`