Add foreground color estimation pass

2026-06-30 16:25:56 +08:00
parent 1a670088c0
commit d6bb658244
9 changed files with 158 additions and 70 deletions
@@ -19,6 +19,10 @@ REQUIRED_DEBUG_FILES = {
    "trimap.png",
    "alpha.png",
    "despill_mask.png",
+    "foreground_mask.png",
+    "foreground_unmix_mask.png",
+    "foreground_local_mask.png",
+    "foreground_rgb.png",
    "preview_black.png",
    "preview_white.png",
    "preview_gray.png",
@@ -62,7 +66,7 @@ def main() -> int:
        default=None,
        help="Override config and allow chroma alpha when ViTMatte fails.",
    )
-    parser.add_argument("--max-edge-green-excess-p95", type=float, default=0.50)
+    parser.add_argument("--max-edge-green-excess-p95", type=float, default=0.30)
    args = parser.parse_args()

    images = sorted(args.samples_dir.glob("*.png"))