Remove stochastic proessor from training pipeline

oarriaga · May 10, 2021 · 30fc2fa · 30fc2fa
1 parent fc42079
commit 30fc2fa
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/examples/object_detection/detection.py b/examples/object_detection/detection.py
@@ -10,10 +10,10 @@ class AugmentImage(SequentialProcessor):
     """
     def __init__(self):
         super(AugmentImage, self).__init__()
-        self.add(pr.Stochastic(pr.RandomContrast(), 0.5))
-        self.add(pr.Stochastic(pr.RandomBrightness(), 0.5))
-        self.add(pr.Stochastic(pr.RandomSaturation(0.7), 0.5))
-        self.add(pr.Stochastic(pr.RandomHue(), 0.5))
+        self.add(pr.RandomContrast())
+        self.add(pr.RandomBrightness())
+        self.add(pr.RandomSaturation(0.7))
+        self.add(pr.RandomHue())
 
 
 class PreprocessImage(SequentialProcessor):