lhk229
62f175948e
Replace the hardcoded green-screen auto-detection with a general flat-colour detector. When screen_color is null the background colour is now detected from the image border (dominant colour of the border strip) and used directly as the chroma model, with a failure gate that raises when the border is not one clean flat colour (gradient / texture / subject filling the frame). - chroma: add detect_background_color + _background_border_cluster; the null case of estimate_background_model samples the detected cluster directly (no seed search); compute_bg_confidence is now always perceptual (Lab/RGB distance). Removed the green heuristic (initial_green_candidates), the green multiplicative gating, the now-unused _smoothstep, and five green-only ChromaSettings fields. - pipeline: reuse the detected colour for de-spill in the auto case. - settings / default.yaml: add detect_* tuning fields; refresh screen_color docs. - README: document auto-detection + the failure gate, trimap modes, birefnet config. A supplied --screen-color hex still uses the seed-search refinement. Validated: detects green / pastel / text samples correctly, raises on gradient / two-colour, CLI exits 1 on failure, green and pastel mattes unchanged in quality. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
69 lines
2.2 KiB
YAML
69 lines
2.2 KiB
YAML
# Background-colour prior. null = auto-detect the flat background colour from the
|
|
# image border (raises if the border is not one clean flat colour). A hex string
|
|
# like "#CFEFFF" sets it explicitly. Also overridable via --screen-color.
|
|
screen_color: null
|
|
|
|
model:
|
|
model_name: hustvl/vitmatte-base-composition-1k
|
|
device: cuda
|
|
matting_method: vitmatte
|
|
fallback_to_chroma_alpha: false
|
|
|
|
segmentation:
|
|
# enabled: true = single-segmenter pipeline (one seg model drives the trimap)
|
|
# enabled: false = no seg model, chroma-only pipeline (Chroma + ViTMatte)
|
|
enabled: true
|
|
backend: anime-seg # single trimap segmenter; switchable: anime-seg | birefnet
|
|
model_name: skytnt/anime-seg # birefnet -> ZhengPeng7/BiRefNet
|
|
device: cuda
|
|
input_size: 1024
|
|
|
|
trimap:
|
|
sure_bg_threshold: 0.92
|
|
sure_fg_threshold: 0.12
|
|
unknown_radius_ratio: 0.012
|
|
fg_safe_radius_ratio: 0.006
|
|
seg_core_threshold: 0.60
|
|
seg_loose_threshold: 0.08
|
|
# Trimap mode (segmentation pipeline only; also settable via --trimap-mode):
|
|
# directional = chroma + seg + hue split; background-hued residual
|
|
# (blue between hair) -> unknown, white shirt kept. (default)
|
|
# seg = original fuse_trimap (no hue split).
|
|
# directional-hard-bg = hard-remove background-hued pixels (aggressive; can eat
|
|
# cool/shadowed white cloth).
|
|
mode: directional
|
|
seg_low: 0.15
|
|
bg_hue_proj_min: 4.0
|
|
|
|
alpha_post:
|
|
chroma_suppress: true
|
|
chroma_suppress_bg_low: 0.35
|
|
chroma_suppress_bg_high: 0.80
|
|
chroma_suppress_strength: 1.0
|
|
|
|
foreground:
|
|
enabled: true
|
|
method: ml
|
|
fallback_to_unmix: true
|
|
ml_regularization: 0.00001
|
|
# Legacy "unmix" method params (used only when method: unmix).
|
|
edge_low: 0.005
|
|
edge_high: 0.995
|
|
min_unmix_alpha: 0.08
|
|
unmix_strength: 0.70
|
|
local_strength: 0.45
|
|
local_blur_radius: 11
|
|
local_alpha_threshold: 0.92
|
|
local_bg_confidence_max: 0.20
|
|
green_excess_margin: 0.015
|
|
bg_confidence_weight: 0.70
|
|
|
|
despill:
|
|
enabled: true
|
|
edge_low: 0.005
|
|
strength: 0.92
|
|
green_excess_margin: 0.015
|
|
edge_expand_radius: 2
|
|
alpha_weight_floor: 0.35
|
|
color_margin_lab: 4.0 # custom-screen despill: Lab chroma margin (screen_color set)
|