Bug Fix

devbruce · Aug 9, 2019 · bf75ad0 · bf75ad0
1 parent c895a54
commit bf75ad0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/CutMixImageDataGenerator.py b/CutMixImageDataGenerator.py
@@ -53,11 +53,11 @@ def __next__(self):
 
         X = X_1
         for i in range(reshape_size):
-            cut_size = int(self.img_size * cut_ratio[i])
-            x1 = random.randint(0, (self.img_size-1) - cut_size)
+            cut_size = int((self.img_size-1) * cut_ratio[i])
             y1 = random.randint(0, (self.img_size-1) - cut_size)
-            x2 = x1 + cut_size
+            x1 = random.randint(0, (self.img_size-1) - cut_size)
             y2 = y1 + cut_size
+            x2 = x1 + cut_size
             cut_arr = cut_img[i][y1:y2, x1:x2]
             cutmix_img = X_1[i]
             cutmix_img[y1:y2, x1:x2] = cut_arr