Merge pull request #1095 from rodrigoberriel/horiz_flip_cfg

Move horizontal flip probability to config
facebookresearch · Oct 17, 2019 · f027259 · f027259
2 parents 523ae86 + ec5ed93
commit f027259
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/maskrcnn_benchmark/config/defaults.py b/maskrcnn_benchmark/config/defaults.py
@@ -60,6 +60,8 @@
 _C.INPUT.SATURATION = 0.0
 _C.INPUT.HUE = 0.0
 
+# Flips
+_C.INPUT.HORIZONTAL_FLIP_PROB_TRAIN = 0.5
 _C.INPUT.VERTICAL_FLIP_PROB_TRAIN = 0.0
 
 # -----------------------------------------------------------------------------

diff --git a/maskrcnn_benchmark/data/transforms/build.py b/maskrcnn_benchmark/data/transforms/build.py
@@ -6,7 +6,7 @@ def build_transforms(cfg, is_train=True):
     if is_train:
         min_size = cfg.INPUT.MIN_SIZE_TRAIN
         max_size = cfg.INPUT.MAX_SIZE_TRAIN
-        flip_horizontal_prob = 0.5  # cfg.INPUT.FLIP_PROB_TRAIN
+        flip_horizontal_prob = cfg.INPUT.HORIZONTAL_FLIP_PROB_TRAIN
         flip_vertical_prob = cfg.INPUT.VERTICAL_FLIP_PROB_TRAIN
         brightness = cfg.INPUT.BRIGHTNESS
         contrast = cfg.INPUT.CONTRAST