Fix alpha_mask transformation

u-haru · u-haru · commit e2034624344f · 2024-03-31T03:46:34.000+09:00
diff --git a/library/train_util.py b/library/train_util.py
@@ -1169,7 +1169,7 @@ def __getitem__(self, index):
                 alpha_mask = cv2.resize(
                     alpha_mask, target_size, interpolation=cv2.INTER_AREA
                 )
-                alpha_mask = self.image_transforms(alpha_mask)
+                alpha_mask = transforms.ToTensor()(alpha_mask)
                 alpha_mask_list.append(alpha_mask)
 
             if not flipped:

Original file line number	Diff line number	Diff line change
`@@ -1169,7 +1169,7 @@ def __getitem__(self, index):`
`1169`	`1169`	`alpha_mask = cv2.resize(`
`1170`	`1170`	`alpha_mask, target_size, interpolation=cv2.INTER_AREA`
`1171`	`1171`	`)`
`1172`		`- alpha_mask = self.image_transforms(alpha_mask)`
	`1172`	`+ alpha_mask = transforms.ToTensor()(alpha_mask)`
`1173`	`1173`	`alpha_mask_list.append(alpha_mask)`
`1174`	`1174`
`1175`	`1175`	`if not flipped:`