Update anime_aesthetic.py

anime_aesthetic.py

@@ -216,11 +216,18 @@ def rescale_pad(image, output_size, random_pad=False):
     if h != output_size or w != output_size:
         r = min(output_size / h, output_size / w)
         new_h, new_w = int(h * r), int(w * r)
+        if random_pad:
+            r2 = random.uniform(0.9, 1)
+            new_h, new_w = int(new_h * r2), int(new_w * r2)
         ph = output_size - new_h
         pw = output_size - new_w
+        left = random.randint(0, pw) if random_pad else pw // 2
+        right = pw - left
+        top = random.randint(0, ph) if random_pad else ph // 2
+        bottom = ph - top
         image = transforms.functional.resize(image, [new_h, new_w])
         image = transforms.functional.pad(
-            image, [pw // 2, ph // 2, pw // 2 + pw % 2, ph // 2 + ph % 2], random.uniform(0, 1) if random_pad else 0
+            image, [left, top, right, bottom], random.uniform(0, 1) if random_pad else 0
         )
     return image
 
@@ -435,7 +442,7 @@ if __name__ == "__main__":
     parser.add_argument(
         "--data-split",
         type=float,
-        default=0.9995,
+        default=0.9999,
         help="split rate for training and validation",
     )
 
@@ -486,7 +493,7 @@ if __name__ == "__main__":
         "--log-step", type=int, default=2, help="log training loss every n steps"
     )
     parser.add_argument(
-        "--val-epoch", type=int, default=0.1, help="valid and save every n epoch"
+        "--val-epoch", type=int, default=0.025, help="valid and save every n epoch"
     )
 
     opt = parser.parse_args()