pymatting 1.1.15: estimate_alpha_lbdm

Clean-room public code sample — generation instructions

Write a brand-new, minimal, self-contained code sample in this clean-room directory.
Do not copy, paraphrase, or reference any existing project source. Work only from this spec.

A csx.json manifest scaffold already exists. Do not recreate it from memory. Preserve its case.goal, packages and symbols; fill its empty case.contract with exact assertions and correct its environment, commands and verifierAdapter for the files you generate.

Goal: verify pymatting.estimate_alpha_lbdm in pkg:pypi/pymatting@1.1.15
Kind: HOW

Use EXACTLY these public packages and versions:
  - pkg:pypi/pymatting@1.1.15
Demonstrate these symbols/APIs:
  - pymatting.estimate_alpha_lbdm

Rules:
  - One focused purpose; the smallest project that proves the goal.
  - Include a contract test (test/contract.*) that runs OFFLINE and exits 0 exactly when the goal behavior works.
  - Pin every dependency with a lockfile so resolution is reproducible.
  - No secrets, credentials, or tokens. No real URLs (only example.com or localhost). No absolute paths.
  - No personal names, emails, company names, or project identifiers of any kind.
  - No binaries and no generated output (node_modules, dist, target, venv, .git, .env).
  - Keep it under 200 files and 256KB packed.

{"case":{"caseId":"case:sha256:1c5e5796eff02faef7e5cd5eb6bdbcf11dc7a9f8b4efde32f78c96755d181987","contract":["assert estimate_alpha_lbdm computes learning-based alpha matte bounded in [0, 1] matching trimap constraints for RGB image","assert estimate_alpha_lbdm accepts laplacian_kwargs and cg_kwargs to configure solver parameters","assert estimate_alpha_lbdm accepts custom preconditioner function","assert estimate_alpha_lbdm raises ValueError when trimap is missing background values","assert estimate_alpha_lbdm raises ValueError when trimap is missing foreground values"],"goal":"verify pymatting.estimate_alpha_lbdm in pkg:pypi/pymatting@1.1.15","kind":"HOW","packages":["pkg:pypi/pymatting@1.1.15"],"schemaVersion":1,"symbols":["pymatting.estimate_alpha_lbdm"]},"contractCommand":["python3","test/contract.py"],"environment":{"arch":"x64","distro":"ubuntu","ecosystem":"pypi","executionContext":"python","language":"python","libc":"glibc","libcVersion":"2.39","moduleSystem":"standard","os":"linux","osVersionBucket":"24","packageManager":"pip","packageManagerVersion":"24.0","runtime":"python","runtimeVersion":"3.12","schemaVersion":1},"license":"MIT-0","packages":["pkg:pypi/pymatting@1.1.15"],"schemaVersion":1,"subject":"pkg:pypi/pymatting@1.1.15","symbols":["pymatting.estimate_alpha_lbdm"],"verifierAdapter":"python@1"}

[project]
name = "verify-pymatting"
version = "0.1.0"
dependencies = [
    "pymatting==1.1.15",
]

pymatting==1.1.15
numpy
scipy
pillow
numba
llvmlite

{
  "schemaVersion": 1,
  "goal": "verify pymatting.estimate_alpha_lbdm in pkg:pypi/pymatting@1.1.15",
  "kind": "HOW",
  "packages": [
    "pkg:pypi/pymatting@1.1.15"
  ],
  "symbols": [
    "pymatting.estimate_alpha_lbdm"
  ]
}

import numpy as np
from pymatting import estimate_alpha_lbdm, ichol, jacobi


def test_estimate_alpha_lbdm_basic():
    h, w = 12, 12
    image = np.zeros((h, w, 3), dtype=np.float64)
    trimap = np.full((h, w), 0.5, dtype=np.float64)

    trimap[:, :4] = 0.0
    trimap[:, 8:] = 1.0

    image[:, :4] = [0.1, 0.1, 0.1]
    image[:, 8:] = [0.9, 0.9, 0.9]
    image[:, 4:8] = [0.5, 0.5, 0.5]

    alpha = estimate_alpha_lbdm(image, trimap)
    assert alpha.shape == (h, w)
    assert np.all(alpha >= 0.0) and np.all(alpha <= 1.0)
    assert np.all(alpha[:, :4] < 0.1)
    assert np.all(alpha[:, 8:] > 0.9)


def test_estimate_alpha_lbdm_solver_parameters():
    h, w = 12, 12
    image = np.zeros((h, w, 3), dtype=np.float64)
    trimap = np.full((h, w), 0.5, dtype=np.float64)

    trimap[:, :4] = 0.0
    trimap[:, 8:] = 1.0

    image[:, :4] = [0.1, 0.1, 0.1]
    image[:, 8:] = [0.9, 0.9, 0.9]
    image[:, 4:8] = [0.5, 0.5, 0.5]

    alpha = estimate_alpha_lbdm(
        image,
        trimap,
        laplacian_kwargs={"epsilon": 1e-6, "radius": 1},
        cg_kwargs={"maxiter": 1000},
    )
    assert alpha.shape == (h, w)
    assert np.all(alpha >= 0.0) and np.all(alpha <= 1.0)


def test_estimate_alpha_lbdm_custom_preconditioner():
    h, w = 12, 12
    image = np.zeros((h, w, 3), dtype=np.float64)
    trimap = np.full((h, w), 0.5, dtype=np.float64)

    trimap[:, :4] = 0.0
    trimap[:, 8:] = 1.0

    image[:, :4] = [0.1, 0.1, 0.1]
    image[:, 8:] = [0.9, 0.9, 0.9]
    image[:, 4:8] = [0.5, 0.5, 0.5]

    alpha_jacobi = estimate_alpha_lbdm(image, trimap, preconditioner=jacobi)
    assert alpha_jacobi.shape == (h, w)
    assert np.all(alpha_jacobi >= 0.0) and np.all(alpha_jacobi <= 1.0)

    alpha_ichol = estimate_alpha_lbdm(image, trimap, preconditioner=ichol)
    assert alpha_ichol.shape == (h, w)
    assert np.all(alpha_ichol >= 0.0) and np.all(alpha_ichol <= 1.0)


def test_estimate_alpha_lbdm_missing_background_raises_value_error():
    h, w = 8, 8
    image = np.ones((h, w, 3), dtype=np.float64) * 0.5
    trimap = np.ones((h, w), dtype=np.float64)

    try:
        estimate_alpha_lbdm(image, trimap)
        assert False, "Should raise ValueError when background is missing"
    except ValueError as exc:
        assert "Trimap did not contain background values" in str(exc)


def test_estimate_alpha_lbdm_missing_foreground_raises_value_error():
    h, w = 8, 8
    image = np.ones((h, w, 3), dtype=np.float64) * 0.5
    trimap = np.zeros((h, w), dtype=np.float64)

    try:
        estimate_alpha_lbdm(image, trimap)
        assert False, "Should raise ValueError when foreground is missing"
    except ValueError as exc:
        assert "Trimap did not contain foreground values" in str(exc)


def main():
    test_estimate_alpha_lbdm_basic()
    test_estimate_alpha_lbdm_solver_parameters()
    test_estimate_alpha_lbdm_custom_preconditioner()
    test_estimate_alpha_lbdm_missing_background_raises_value_error()
    test_estimate_alpha_lbdm_missing_foreground_raises_value_error()
    print("All contract assertions passed.")


if __name__ == "__main__":
    main()