fix

nguyenhongchau · Aug 9, 2018 · 2e9daf6 · 2e9daf6
1 parent e07493f
commit 2e9daf6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/age_estimation/generator.py b/age_estimation/generator.py
@@ -13,7 +13,11 @@ def get_transform_func():
     p = Augmentor.Pipeline()
     p.flip_left_right(probability=0.5)
     p.rotate(probability=1, max_left_rotation=5, max_right_rotation=5)
-    p.zoom_random(probability=1, percentage_area=0.95)
+    p.zoom_random(probability=0.5, percentage_area=0.95)
+    p.random_distortion(probability=0.5, grid_width=2, grid_height=2, magnitude=8)
+    p.random_color(probability=1, min_factor=0.8, max_factor=1.2)
+    p.random_contrast(probability=1, min_factor=0.8, max_factor=1.2)
+    p.random_brightness(probability=1, min_factor=0.8, max_factor=1.2)
 
     def transform_image(image):
         image = [Image.fromarray(image)]