refactor: Minor refactor for probability

yakhyo · Oct 18, 2024 · 6105133 · 6105133
1 parent a166d34
commit 6105133
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/transform.py b/utils/transform.py
@@ -102,7 +102,7 @@ def apply_adjustments(image, alpha=1.0, beta=0.0):
         apply_adjustments(image[:, :, 1], alpha=random.uniform(0.5, 1.5))
 
     # Hue distortion
-    if random.choice([True, False]):
+    if random.random() >= 0.5:
         hue_shift = image[:, :, 0].astype(int) + random.randint(-18, 18)
         image[:, :, 0] = np.mod(hue_shift, 180)