add tunable parameters for CEH module

Author: QiuJueqin

configs/nikon_d3x.yaml

@@ -10,6 +10,7 @@ module_enable_status:
   csc: True
   nlm: False
   bnf: True
+  ceh: True
   eeh: True
   fcs: True
   hsc: True
@@ -73,6 +74,10 @@ bnf:
   intensity_sigma: 0.8                # larger value has stronger smoothing effect
   spatial_sigma: 0.8                  # larger value has stronger smoothing effect
 
+ceh:
+  tiles: [4, 6]
+  clip_limit: 0.02
+
 eeh:
   flat_threshold: 4                   # delta <= flat_threshold: set delta to 0
   edge_threshold: 8                   # delta > edge_threshold: increase delta by edge_gain

configs/test.yaml

@@ -10,6 +10,7 @@ module_enable_status:
   csc: True
   nlm: True
   bnf: True
+  ceh: True
   eeh: True
   fcs: True
   hsc: True
@@ -73,6 +74,10 @@ bnf:
   intensity_sigma: 0.8                # larger value has stronger smoothing effect
   spatial_sigma: 0.8                  # larger value has stronger smoothing effect
 
+ceh:
+  tiles: [4, 6]
+  clip_limit: 0.01
+
 eeh:
   flat_threshold: 4                   # delta <= flat_threshold: set delta to 0
   edge_threshold: 8                   # delta > edge_threshold: increase delta by edge_gain

demo.py

@@ -49,11 +49,12 @@ def demo_intermediate_results():
     with cfg.unfreeze():
         cfg.module_enable_status.nlm = True
         cfg.nlm.h = 15
+        cfg.ceh.clip_limit = 0.04
         cfg.eeh.flat_threshold = 1
         cfg.eeh.flat_threshold = 2
         cfg.eeh.edge_gain = 1280
-        cfg.hsc.saturation_gain = 384
-        cfg.bcc.contrast_gain = 300
+        cfg.hsc.saturation_gain = 320
+        cfg.bcc.contrast_gain = 280
 
     pipeline = Pipeline(cfg)