diffengine.datasets.transforms.dump_image

Module Contents

Classes

DumpImage

Dump the image processed by the pipeline.

DumpMaskedImage

Dump Masked the image processed by the pipeline.

class diffengine.datasets.transforms.dump_image.DumpImage(max_imgs, dump_dir)[source]

Dump the image processed by the pipeline.

Args:

max_imgs (int): Maximum value of output. dump_dir (str): Dump output directory.

__call__(results)[source]

Dump the input image to the specified directory.

No changes will be made.

Args:

results (dict): Result dict from loading pipeline.

Returns:

results (dict): Result dict from loading pipeline. (same as input)

Parameters:

results (dict) –

Return type:

dict

Parameters:
  • max_imgs (int) –

  • dump_dir (str) –

class diffengine.datasets.transforms.dump_image.DumpMaskedImage(max_imgs, dump_dir)[source]

Dump Masked the image processed by the pipeline.

Args:

max_imgs (int): Maximum value of output. dump_dir (str): Dump output directory.

__call__(results)[source]

Dump the input image to the specified directory.

No changes will be made.

Args:

results (dict): Result dict from loading pipeline.

Returns:

results (dict): Result dict from loading pipeline. (same as input)

Parameters:

results (dict) –

Return type:

dict

Parameters:
  • max_imgs (int) –

  • dump_dir (str) –