Add foreground color estimation pass

bgfilter/settings.py

@@ -35,6 +35,21 @@ class AlphaPostSettings:
     alpha_ceil: float = 0.998
 
 
+@dataclass(frozen=True)
+class ForegroundSettings:
+    enabled: bool = True
+    edge_low: float = 0.005
+    edge_high: float = 0.995
+    min_unmix_alpha: float = 0.08
+    unmix_strength: float = 0.70
+    local_strength: float = 0.45
+    local_blur_radius: int = 11
+    local_alpha_threshold: float = 0.92
+    local_bg_confidence_max: float = 0.20
+    green_excess_margin: float = 0.015
+    bg_confidence_weight: float = 0.70
+
+
 @dataclass(frozen=True)
 class DespillSettings:
     enabled: bool = True
@@ -45,12 +60,6 @@ class DespillSettings:
     edge_expand_radius: int = 2
     bg_confidence_weight: float = 0.65
     alpha_weight_floor: float = 0.35
-    unmix_strength: float = 0.55
-    unmix_alpha_min: float = 0.08
-    local_fg_strength: float = 0.35
-    local_fg_blur_radius: int = 9
-    local_fg_alpha_threshold: float = 0.92
-    local_fg_bg_confidence_max: float = 0.20
 
 
 @dataclass(frozen=True)
@@ -66,5 +75,6 @@ class PipelineSettings:
     chroma: ChromaSettings = ChromaSettings()
     trimap: TrimapSettings = TrimapSettings()
     alpha_post: AlphaPostSettings = AlphaPostSettings()
+    foreground: ForegroundSettings = ForegroundSettings()
     despill: DespillSettings = DespillSettings()
     model: ModelSettings = ModelSettings()