Shortcuts

mmedit.apis.inferencers.video_restoration_inferencer

Module Contents

Classes

VideoRestorationInferencer

inferencer that predicts with video restoration models.

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

Bases: mmedit.apis.inferencers.base_mmedit_inferencer.BaseMMEditInferencer

inferencer that predicts with video restoration models.

func_kwargs[source]
extra_parameters[source]
preprocess(video: mmedit.apis.inferencers.base_mmedit_inferencer.InputsType) Dict[source]

Process the inputs into a model-feedable format.

Parameters

video (InputsType) – Video to be restored by models.

Returns

Results of preprocess.

Return type

results(InputsType)

forward(inputs: mmedit.apis.inferencers.base_mmedit_inferencer.InputsType) mmedit.apis.inferencers.base_mmedit_inferencer.PredType[source]

Forward the inputs to the model.

Parameters

inputs (InputsType) – Images array of input video.

Returns

Results of forwarding

Return type

PredType

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

Visualize predictions.

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

  • data (List[Dict]) – Not needed by this kind of inferencer.

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

Returns

Result of visualize

Return type

List[np.ndarray]

postprocess(preds: mmedit.apis.inferencers.base_mmedit_inferencer.PredType, imgs: Optional[List[numpy.ndarray]] = None) Union[mmedit.apis.inferencers.base_mmedit_inferencer.ResType, Tuple[mmedit.apis.inferencers.base_mmedit_inferencer.ResType, numpy.ndarray]][source]

Postprocess is not needed in this inferencer.

Read the Docs v: latest
Versions
master
latest
stable
zyh-re-docs
zyh-doc-notfound-extend
zyh-api-rendering
Downloads
pdf
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.