add vibrance tuning

app.py

@@ -73,12 +73,26 @@ def tone_map(
     return hsv
 
 
-@interactive(vibrance=(0, [0, 100]))
-def modify_vibrance(hsv_in, vibrance: float = 0.):
-    gain_luma = np.interp(hsv_in[..., 2], [0., 0.5, 1.], [0., 0., 1.])
-    gain_sat = np.interp(hsv_in[..., 1], [0., 0.5, 1.], [0., 0., 1.])**2
-    red_thresh = 30.
-    pink_thresh = 8.
+@interactive(
+    vibrance=(0, [0, 100]),
+    # vibrance_mask_smoothness=(3., [-0.9, 4.])
+)
+def modify_vibrance(
+    hsv_in: np.ndarray,
+    vibrance: float = 0.,
+    vibrance_mask_smoothness: float = 3.
+) -> np.ndarray:
+    gain_luma = np.interp(
+        hsv_in[..., 2],
+        [0., 0.5, 1.],
+        [0., 0., 1.]
+    )
+    gain_sat = np.interp(hsv_in[..., 1],
+                         [0., 0.25, 1.],
+                         [0., 0., 1.]
+                         )**(1+vibrance_mask_smoothness)
+    red_thresh = 55.
+    pink_thresh = 10.
     overlap = 10.
 
     gain_hue = np.interp(
@@ -86,8 +100,8 @@ def modify_vibrance(hsv_in, vibrance: float = 0.):
         [0., red_thresh, red_thresh+overlap,
          360-pink_thresh-overlap, 360-pink_thresh, 360],
         [0., 0., 1., 1., 0., 0.]
-    )**2
-    gain = 1. + gain_luma * gain_sat * gain_hue*(vibrance / 100)
+    )**(1+vibrance_mask_smoothness)
+    gain = 1. + gain_luma * (gain_sat + gain_hue).clip(0., 1.)*(vibrance / 100)
     hsv_in[..., 1] = np.clip(gain*hsv_in[..., 1], 0, 1)
     return hsv_in