mmagic.datasets.transforms.random_degradations

Module Contents

Classes

RandomBlur	Apply random blur to the input.
RandomJPEGCompression	Apply random JPEG compression to the input.
RandomNoise	Apply random noise to the input.
RandomResize	Randomly resize the input.
RandomVideoCompression	Apply random video compression to the input.
DegradationsWithShuffle	Apply random degradations to input, with degradations being shuffled.

Attributes

has_av
allowed_degradations

mmagic.datasets.transforms.random_degradations.has_av = True

class mmagic.datasets.transforms.random_degradations.RandomBlur(params, keys)

Apply random blur to the input.

Modified keys are the attributed specified in “keys”.

参数

• params (dict) – A dictionary specifying the degradation settings.

• keys (list[str]) – A list specifying the keys whose values are modified.

get_kernel(num_kernels: int)

This is the function to create kernel.

参数

num_kernels (int) – the number of kernels

返回

_description_

返回类型

_type_

_apply_random_blur(imgs)

This is the function to apply blur operation on images.

参数

imgs (Tensor) – images

返回

Images applied blur

返回类型

Tensor

__call__(results)

Call this transform.

__repr__()

Print the basic information of the transform.

class mmagic.datasets.transforms.random_degradations.RandomJPEGCompression(params, keys, color_type='color', bgr2rgb=False)

Apply random JPEG compression to the input.

Modified keys are the attributed specified in “keys”.

参数

• params (dict) – A dictionary specifying the degradation settings.

• keys (list[str]) – A list specifying the keys whose values are modified.

• bgr2rgb (str) – Whether change channel order. Default: False.

_apply_random_compression(imgs)

__call__(results)

Call this transform.

__repr__()

Print the basic information of the transform.

class mmagic.datasets.transforms.random_degradations.RandomNoise(params, keys)

Apply random noise to the input.

Currently support Gaussian noise and Poisson noise.

Modified keys are the attributed specified in “keys”.

参数

• params (dict) – A dictionary specifying the degradation settings.

• keys (list[str]) – A list specifying the keys whose values are modified.

_apply_gaussian_noise(imgs)

This is the function used to apply gaussian noise on images.

参数

imgs (Tensor) – images

返回

images applied gaussian noise

返回类型

Tensor

_apply_poisson_noise(imgs)

_apply_random_noise(imgs)

This is the function used to apply random noise on images.

参数

imgs (Tensor) – training images

返回

_description_

返回类型

_type_

__call__(results)

Call this transform.

__repr__()

Print the basic information of the transform.

class mmagic.datasets.transforms.random_degradations.RandomResize(params, keys)

Randomly resize the input.

Modified keys are the attributed specified in “keys”.

参数

• params (dict) – A dictionary specifying the degradation settings.

• keys (list[str]) – A list specifying the keys whose values are modified.

_random_resize(imgs)

This is the function used to randomly resize images for training augmentation.

参数

imgs (Tensor) – training images.

返回

images after randomly resized

返回类型

Tensor

__call__(results)

Call this transform.

__repr__()

Print the basic information of the transform.

class mmagic.datasets.transforms.random_degradations.RandomVideoCompression(params, keys)

Apply random video compression to the input.

Modified keys are the attributed specified in “keys”.

参数

• params (dict) – A dictionary specifying the degradation settings.

• keys (list[str]) – A list specifying the keys whose values are modified.

_apply_random_compression(imgs)

This is the function to apply random compression on images.

参数

imgs (Tensor) – training images

返回

images after randomly compressed

返回类型

Tensor

__call__(results)

Call this transform.

__repr__()

Print the basic information of the transform.

mmagic.datasets.transforms.random_degradations.allowed_degradations

class mmagic.datasets.transforms.random_degradations.DegradationsWithShuffle(degradations, keys, shuffle_idx=None)

Apply random degradations to input, with degradations being shuffled.

Degradation groups are supported. The order of degradations within the same group is preserved. For example, if we have degradations = [a, b, [c, d]] and shuffle_idx = None, then the possible orders are

[a, b, [c, d]]
[a, [c, d], b]
[b, a, [c, d]]
[b, [c, d], a]
[[c, d], a, b]
[[c, d], b, a]

Modified keys are the attributed specified in “keys”.

参数

• degradations (list[dict]) – The list of degradations.

• keys (list[str]) – A list specifying the keys whose values are modified.

• shuffle_idx (list | None, optional) – The degradations corresponding to these indices are shuffled. If None, all degradations are shuffled. Default: None.

_build_degradations(degradations)

__call__(results)

Call this transform.

__repr__()

Print the basic information of the transform.