Method __call__

libcom/os_insert/os_insert.py:137–313

Run a single OSInsert inference. Parameters ---------- background_path: Path to the background image. foreground_path: Path to the foreground image used as the InsertAnything reference image. foreground_mask_path:

(
        self,
        background_path: str | Path,
        foreground_path: str | Path,
        foreground_mask_path: str | Path,
        bbox_txt_path: str | Path,
        result_dir: str | Path,
        mode: Literal["aggressive", "conservative"] = "conservative",
        cleanup_intermediate: bool = True,
        verbose: bool = False,
        seed: int = 123,
        strength: float = 1.0,
        split_ratio: float = 0.5,
    )

        return self._sam_predictor

    def __call__(
        self,
        background_path: str | Path,
        foreground_path: str | Path,
        foreground_mask_path: str | Path,
        bbox_txt_path: str | Path,
        result_dir: str | Path,
        mode: Literal["aggressive", "conservative"] = "conservative",
        cleanup_intermediate: bool = True,
        verbose: bool = False,
        seed: int = 123,
        strength: float = 1.0,
        split_ratio: float = 0.5,
    ) -> np.ndarray | None:
        """Run a single OSInsert inference.

        Parameters
        ----------
        background_path:
            Path to the background image.
        foreground_path:
            Path to the foreground image used as the InsertAnything reference
            image.
        foreground_mask_path:
            Binary mask for the foreground image.
        bbox_txt_path:
            Text file containing ``x1 y1 x2 y2`` on a single line, specifying the
            insertion region on the background image.
        result_dir:
            Directory where the final composed image will be written.
        mode:
            - ``"conservative"``: background + bbox -> mask -> InsertAnything.
            - ``"aggressive"``: ObjectStitch + SAM -> combined source/mask ->
              InsertAnything.
        cleanup_intermediate:
            Deprecated. Present for backward compatibility.
        verbose:
            If True, save intermediate artifacts into ``result_dir/intermediates``.
            Default False (do not save intermediates).
        seed:
            Random seed for InsertAnything.
        strength:
            InsertAnything strength parameter.
        """

        if mode not in {"aggressive", "conservative"}:
            raise ValueError(f"Unsupported mode: {mode}")

        # ------------------------------------------------------------------
        # Path normalization and output directory.
        # ------------------------------------------------------------------
        background_path = Path(background_path)
        foreground_path = Path(foreground_path)
        foreground_mask_path = Path(foreground_mask_path)
        bbox_txt_path = Path(bbox_txt_path)
        result_dir = Path(result_dir)

        os.makedirs(result_dir, exist_ok=True)