mirror of
https://github.com/open-mmlab/mmpretrain.git
synced 2025-06-03 14:59:18 +08:00
09597e5a4c
* Add transform `RandomErasing`. * Add unittests of `RandomErasing` * Fix typo in docstring * Improve docstring and unittests.
22 lines
1.2 KiB
Python
22 lines
1.2 KiB
Python
from .auto_augment import (AutoAugment, AutoContrast, Brightness,
|
|
ColorTransform, Contrast, Cutout, Equalize, Invert,
|
|
Posterize, RandAugment, Rotate, Sharpness, Shear,
|
|
Solarize, SolarizeAdd, Translate)
|
|
from .compose import Compose
|
|
from .formating import (Collect, ImageToTensor, ToNumpy, ToPIL, ToTensor,
|
|
Transpose, to_tensor)
|
|
from .loading import LoadImageFromFile
|
|
from .transforms import (CenterCrop, ColorJitter, Lighting, RandomCrop,
|
|
RandomErasing, RandomFlip, RandomGrayscale,
|
|
RandomResizedCrop, Resize)
|
|
|
|
__all__ = [
|
|
'Compose', 'to_tensor', 'ToTensor', 'ImageToTensor', 'ToPIL', 'ToNumpy',
|
|
'Transpose', 'Collect', 'LoadImageFromFile', 'Resize', 'CenterCrop',
|
|
'RandomFlip', 'Normalize', 'RandomCrop', 'RandomResizedCrop',
|
|
'RandomGrayscale', 'Shear', 'Translate', 'Rotate', 'Invert',
|
|
'ColorTransform', 'Solarize', 'Posterize', 'AutoContrast', 'Equalize',
|
|
'Contrast', 'Brightness', 'Sharpness', 'AutoAugment', 'SolarizeAdd',
|
|
'Cutout', 'RandAugment', 'Lighting', 'ColorJitter', 'RandomErasing'
|
|
]
|