mmagic.datasets.transforms.albu_function¶
Module Contents¶
Classes¶
PairedAlbuTransForms augmentation. |
|
AlbuTransForms augmentation. |
|
PairedAlbuNormalize augmentation. |
|
AlbuNormalize augmentation. |
|
AlbuCorruptFunction augmentation. |
Functions¶
|
- class mmagic.datasets.transforms.albu_function.PairedAlbuTransForms(size: int, lq_key: str = 'img', gt_key: str = 'gt', scope: str = 'geometric', crop: str = 'random', p: float = 0.5)[source]¶
Bases:
mmcv.transforms.BaseTransformPairedAlbuTransForms augmentation.
Apply the same AlbuTransforms augmentation to paired images.
- class mmagic.datasets.transforms.albu_function.AlbuTransForms(size: int, keys: List, scope: str = 'geometric', crop: str = 'random', p: float = 0.5)[source]¶
Bases:
mmcv.transforms.BaseTransformAlbuTransForms augmentation.
Apply the same AlbuTransForms augmentation to the input images.
- class mmagic.datasets.transforms.albu_function.PairedAlbuNormalize(lq_key: str, gt_key: str, mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), max_pixel_value: float = 255.0, always_apply: bool = False, p: float = 1.0)[source]¶
Bases:
mmcv.transforms.BaseTransformPairedAlbuNormalize augmentation.
Apply the same AlbuNormalize augmentation to the paired images.
- class mmagic.datasets.transforms.albu_function.AlbuNormalize(keys: List, mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), max_pixel_value: float = 255.0, always_apply: bool = False, p: float = 1.0)[source]¶
Bases:
mmcv.transforms.BaseTransformAlbuNormalize augmentation.
Apply the same AlbuNormalize augmentation to the input images.
- class mmagic.datasets.transforms.albu_function.AlbuCorruptFunction(keys: List[str], config: List[dict], p: float = 1.0)[source]¶
Bases:
mmcv.transforms.BaseTransformAlbuCorruptFunction augmentation.
Apply the same AlbuCorruptFunction augmentation to the input images.