diff --git a/aug.py b/aug.py
new file mode 100644
index 0000000..5da4106
--- /dev/null
+++ b/aug.py
@@ -0,0 +1,32 @@
+import tensorflow as tf
+import numpy as np
+from PIL import Image
+from os import listdir as ls
+from sys import argv, exit
+from tensorflow.keras.utils import array_to_img as img
+
+if len(argv) != 3:
+    exit(1)
+
+files = ls(argv[1])
+
+for v, fn in enumerate(files):
+    i = np.array(Image.open(f"{argv[1]}/{fn}").convert('RGB').resize((300,300))) / 255.0
+    print(f"Processing {v+1:03d}/{len(files)}: {fn}")
+
+    for x1 in range(4):
+        i1 = tf.image.random_brightness(i, 0.6)
+        for x2 in range(4):
+            i2 = tf.image.random_saturation(i1, 0.1, 2.0)
+            for x3 in range(4):
+                i3 = tf.image.random_contrast(i2, 0.2, 1.9)
+                img(i3).save(f"{argv[2]}/{fn.rsplit('.', 1)[0]}_{x1}_{x2}_{x3}_1.png")
+
+                i3 = tf.image.flip_left_right(i3)
+                img(i3).save(f"{argv[2]}/{fn.rsplit('.', 1)[0]}_{x1}_{x2}_{x3}_2.png")
+
+                i3 = tf.image.flip_up_down(i3)
+                img(i3).save(f"{argv[2]}/{fn.rsplit('.', 1)[0]}_{x1}_{x2}_{x3}_3.png")
+
+                i3 = tf.image.flip_left_right(i3)
+                img(i3).save(f"{argv[2]}/{fn.rsplit('.', 1)[0]}_{x1}_{x2}_{x3}_4.png")