random erasing

han-han · Aug 11, 2018 · febcfa7 · febcfa7
1 parent 2e9daf6
commit febcfa7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/age_estimation/generator.py b/age_estimation/generator.py
@@ -18,6 +18,7 @@ def get_transform_func():
     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)
+    p.random_erasing(probability=0.5, rectangle_area=0.2)
 
     def transform_image(image):
         image = [Image.fromarray(image)]