data_juicer.ops.mapper.image_remove_background_mapper module

class data_juicer.ops.mapper.image_remove_background_mapper.ImageRemoveBackgroundMapper(alpha_matting: bool = False, alpha_matting_foreground_threshold: int = 240, alpha_matting_background_threshold: int = 10, alpha_matting_erode_size: int = 10, bgcolor: Tuple[int, int, int, int] | None = None, save_dir: str = None, *args, **kwargs)

基类：Mapper

Mapper to remove background of images

__init__(alpha_matting: bool = False, alpha_matting_foreground_threshold: int = 240, alpha_matting_background_threshold: int = 10, alpha_matting_erode_size: int = 10, bgcolor: Tuple[int, int, int, int] | None = None, save_dir: str = None, *args, **kwargs)

Initialization method.

参数:

• alpha_matting -- (bool, optional) Flag indicating whether to use alpha matting. Defaults to False.

• alpha_matting_foreground_threshold -- (int, optional) Foreground threshold for alpha matting. Defaults to 240.

• alpha_matting_background_threshold -- (int, optional) Background threshold for alpha matting. Defaults to 10.

• alpha_matting_erode_size -- (int, optional) Erosion size for alpha matting. Defaults to 10.

• bgcolor -- (Optional[Tuple[int, int, int, int]], optional) Background color for the cutout image. Defaults to None.

• save_dir -- The directory where generated image files will be stored. If not specified, outputs will be saved in the same directory as their corresponding input files. This path can alternatively be defined by setting the DJ_PRODUCED_DATA_DIR environment variable.

*args (Optional[Any]): Additional positional arguments. **kwargs (Optional[Any]): Additional keyword arguments.

process_single(sample, context=False)

For sample level, sample --> sample

参数:

sample -- sample to process

返回:

processed sample