mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
5195932952
* support random augmentation * minor fix on posterize * minor fix on posterize * minor fix on cutout * minor fix on cutout * fix bug in solarize add * revised according to comments
22 lines
1.1 KiB
Python
22 lines
1.1 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,
|
|
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'
|
|
]
|