Function run_birefnet

clip_manager.py:306–361

(
    clips,
    device=None,
    usage="General",
    dilate_radius=0,
    *,
    on_clip_start: Callable[[str, int], None] | None = None,
    on_frame_complete: Callable[[int, int], None] | None = None,
)

def run_birefnet(
    clips,
    device=None,
    usage="General",
    dilate_radius=0,
    *,
    on_clip_start: Callable[[str, int], None] | None = None,
    on_frame_complete: Callable[[int, int], None] | None = None,
):
    clips_to_process = [c for c in clips if c.alpha_asset is None]

    if not clips_to_process:
        logger.info("All clips have valid Alpha assets. No BiRefNet generation needed.")
        return

    if device is None:
        device = resolve_device()

    logger.info(f"Found {len(clips_to_process)} clips missing Alpha.")

    logger.info(f"Initializing BiRefNet ({usage}) on {device}...")
    # Initialize the handler once
    try:
        handler = BiRefNetHandler(device=device, usage=usage)
    except ImportError as e:
        logger.error(f"BiRefNet Import Error: {e}")
        return
    except Exception as e:
        logger.error(f"BiRefNet Initialization Error: {e}")
        return

    try:
        for clip in clips_to_process:
            logger.info(f"Generating BiRefNet Alpha for: {clip.name}")
            if on_clip_start:
                on_clip_start(clip.name, clip.input_asset.frame_count)

            alpha_output_dir = os.path.join(clip.root_path, "AlphaHint")
            os.makedirs(alpha_output_dir, exist_ok=True)

            try:
                handler.process(
                    input_path=clip.input_asset.path,
                    alpha_output_dir=alpha_output_dir,
                    dilate_radius=dilate_radius,
                    on_frame_complete=on_frame_complete,
                )
                logger.info(f"BiRefNet complete for {clip.name}")
            except Exception as e:
                logger.error(f"BiRefNet failed for {clip.name}: {e}")
                import traceback

                traceback.print_exc()

    finally:
        handler.cleanup()