Fix alpha mask flipping logic

u-haru · u-haru · commit d6484b1db09d · 2024-03-30T05:14:36.000+09:00
diff --git a/library/train_util.py b/library/train_util.py
@@ -1165,7 +1165,7 @@ def __getitem__(self, index):
 
             if subset.alpha_mask and alpha_mask is not None:
                 if flipped:
-                    alpha_mask = alpha_mask[:, ::-1, :].copy()
+                    alpha_mask = alpha_mask[:, ::-1].copy()
                 alpha_mask = cv2.resize(
                     alpha_mask, target_size, interpolation=cv2.INTER_AREA
                 )

Original file line number	Diff line number	Diff line change
`@@ -1165,7 +1165,7 @@ def __getitem__(self, index):`
`1165`	`1165`
`1166`	`1166`	`if subset.alpha_mask and alpha_mask is not None:`
`1167`	`1167`	`if flipped:`
`1168`		`- alpha_mask = alpha_mask[:, ::-1, :].copy()`
	`1168`	`+ alpha_mask = alpha_mask[:, ::-1].copy()`
`1169`	`1169`	`alpha_mask = cv2.resize(`
`1170`	`1170`	`alpha_mask, target_size, interpolation=cv2.INTER_AREA`
`1171`	`1171`	`)`