mmedit.apis.inferencers.inpainting_inferencer

Module Contents

Classes

InpaintingInferencer

inferencer that predicts with inpainting models.

class mmedit.apis.inferencers.inpainting_inferencer.InpaintingInferencer(config: Union[mmedit.utils.ConfigType, str], ckpt: Optional[str], device: Optional[str] = None, extra_parameters: Optional[Dict] = None, seed: int = 2022, **kwargs)

Bases: mmedit.apis.inferencers.base_mmedit_inferencer.BaseMMEditInferencer

inferencer that predicts with inpainting models.

func_kwargs

_init_pipeline(cfg) → mmengine.dataset.Compose

Initialize the test pipeline.

preprocess(img: mmedit.apis.inferencers.base_mmedit_inferencer.InputsType, mask: mmedit.apis.inferencers.base_mmedit_inferencer.InputsType) → Dict

Process the inputs into a model-feedable format.

参数

• img (InputsType) – Image to be inpainted by models.

• mask (InputsType) – Image mask for inpainting models.

返回

Results of preprocess.

返回类型

results(Dict)

forward(inputs: mmedit.apis.inferencers.base_mmedit_inferencer.InputsType) → mmedit.apis.inferencers.base_mmedit_inferencer.PredType

Forward the inputs to the model.

visualize(preds: mmedit.apis.inferencers.base_mmedit_inferencer.PredType, result_out_dir: str = None) → List[numpy.ndarray]

Visualize predictions.

参数

• preds (List[Union[str, np.ndarray]]) – Forward results by the inferencer.

• data (List[Dict]) – Mask of input image.

• result_out_dir (str) – Output directory of image. Defaults to ‘’.

返回

Result of visualize

返回类型

List[np.ndarray]