Class method to return a string value. This method returns the string value 'sam'. Parameters: cls: The class o

Create a new session object based on the specified model name. This function searches for the session class based on the model name in the '

Remove the background from an input image. This function takes in various parameters and returns a modified version of the input image with

Class method to download ONNX model files. This method is responsible for downloading two ONNX model files from specified URLs and s

Perform a simple cutout operation on an image using a mask. This function takes a PIL image `img` and a PIL image `mask` as input. It us

Apply the specified mask to the image as an alpha cutout. Args: img (PILImage): The image to be modified. mask (PILImage): T

Apply affine transformation to an image (matching cv2.warpAffine behavior). cv2.warpAffine maps source coordinates to destination coordinate

Perform alpha matting on an image using a given mask and threshold values. This function takes a PIL image `img` and a PIL image `mask` as i

Apply the specified background color to the image. Args: img (PILImage): The image to be modified. color (Tuple[int, int, in

Download models for image processing.

Fix the orientation of the image based on its EXIF data. Args: img (PILImage): The image to be fixed. Returns: PILImage

Concatenate two images vertically. Args: img1 (PILImage): The first image. img2 (PILImage): The second image to be concatena

Concatenate multiple images vertically. Args: imgs (List[PILImage]): The list of images to be concatenated. Returns: PI

Returns the name of the U2net session. Parameters: *args: Additional positional arguments. **kwargs: Additio

Returns the name of the BiRefNet-COD session. Parameters: *args: Additional positional arguments. **kwargs:

Get the name of the pre-trained model. Parameters: *args: Variable length argument list. **kwargs: Arbitrary

Returns the name of the BiRefNet-Portrait session. Parameters: *args: Additional positional arguments. **kwa

Returns the name of the U2netp model. Returns: str: The name of the model.

Returns the name of the BiRefNet-HRSOD session. Parameters: *args: Additional positional arguments. **kwargs

Returns the name of the Bria-rmbg session. Parameters: *args: Additional positional arguments. **kwargs: Add

Return the name of the model. This method returns the name of the Silueta model. Parameters: *args: Variable le

Returns the name of the BiRefNet-General session. Parameters: *args: Additional positional arguments. **kwar

Returns the name of the BiRefNet-DIS session. Parameters: *args: Additional positional arguments. **kwargs:

Returns the name of the BiRefNet-Massive session. Parameters: *args: Additional positional arguments. **kwar

Returns the name of the BiRefNet-General-Lite session. Parameters: *args: Additional positional arguments. *

Returns the name of the model. This class method returns the name of the model. Parameters: args: Additional po

Returns the name of the U2Net model. Parameters: *args: Variable length argument list. **kwargs: Arbitrary k

Post Process the mask for a smooth boundary by applying Morphological Operations Research based on paper: https://www.sciencedirect.com/scien

Predict masks for an input image. This function takes an image as input and performs various preprocessing steps on the image. It th

Initialize a new BenCustomSession object. Parameters: model_name (str): The name of the model. sess_opts: Th

Initialize an instance of the BaseSession class.

Initialize a new DisCustomSession object. Parameters: model_name (str): The name of the model. sess_opts: Th

Initialize a new SamSession with the given model name and session options. Args: model_name (str): The name of the model

Initialize a new U2netCustomSession object. Parameters: model_name (str): The name of the model. sess_opts (

Command-line interface for processing images by removing the background using a specified model and generating a mask. This CLI command take

Download models

Downloads the U2net model file from a specific URL and saves it. Parameters: *args: Additional positional arguments.

Downloads the BiRefNet-COD model file from a specific URL and saves it. Parameters: *args: Additional positional argumen

Download the pre-trained models. Parameters: *args: Variable length argument list. **kwargs: Arbitrary keywo

Downloads the BiRefNet-Portrait model file from a specific URL and saves it. Parameters: *args: Additional positional ar

Downloads the U2netp model. Returns: str: The path to the downloaded model.

Downloads the BiRefNet-HRSOD model file from a specific URL and saves it. Parameters: *args: Additional positional argum

Download the model files. Parameters: *args: Additional positional arguments. **kwargs: Additional keyword a

Downloads the BRIA-RMBG 2.0 model file from a specific URL and saves it. Parameters: *args: Additional positional argume

Download the pre-trained model file. This method downloads the pre-trained model file from a specified URL. The file is saved to the

Download the model files. Parameters: *args: Additional positional arguments. **kwargs: Additional keyword a

Downloads the BiRefNet-General model file from a specific URL and saves it. Parameters: *args: Additional positional arg

Downloads the BiRefNet-DIS model file from a specific URL and saves it. Parameters: *args: Additional positional argumen

Downloads the BiRefNet-Massive model file from a specific URL and saves it. Parameters: *args: Additional positional arg

Downloads the BiRefNet-General-Lite model file from a specific URL and saves it. Parameters: *args: Additional positiona

Downloads the pre-trained model file. This class method downloads the pre-trained model file from a specified URL using the pooch li

Download the model files. Parameters: *args: Additional positional arguments. **kwargs: Additional keyword a

Downloads the U2Net model weights. Parameters: *args: Variable length argument list. **kwargs: Arbitrary key

Click command line interface function to process an input file based on the provided options. This function is the entry point for the CLI p

Get the name of the model. Parameters: *args: Additional positional arguments. **kwargs: Additional keyword

Get the name of the model. Parameters: *args: Additional positional arguments. **kwargs: Additional keyword

Get the name of the model. Parameters: *args: Additional positional arguments. **kwargs: Additional keyword

Command-line interface (CLI) program for performing background removal on images in a folder. This program takes a folder as input and uses

Predicts the output masks for the input image using the inner session. Parameters: img (PILImage): The input image.

Predict the cloth category of an image. This method takes an image as input and predicts the cloth category of the image. Th

Use a pre-trained model to predict the object in the given image. Parameters: img (PILImage): The input image.

Predicts the mask for the given image using the U2netp model. Parameters: img (PILImage): The input image. Retu

Predicts the mask image for the input image. This method takes a PILImage object as input and returns a list of PILImage objects as

Predicts the output masks for the input image using the inner session. Parameters: img (PILImage): The input image.

Predict the mask of the input image. This method takes an image as input, preprocesses it, and performs a prediction to generate a m

Predicts the mask image for the input image. This method takes a PILImage object as input and returns a list of PILImage objects as

Predicts the output masks for the input image using the inner session. Parameters: img (PILImage): The input image.

Predicts the mask image for the input image. This method takes a PILImage object as input and returns a list of PILImage objects as

Predict the segmentation mask for the input image. Parameters: img (PILImage): The input image. *args: Addit

Predicts human segmentation masks for the input image. Parameters: img (PILImage): The input image. *args: V