2021-08-17 19:52:42 +08:00
|
|
|
# Copyright (c) OpenMMLab. All rights reserved.
|
2021-03-30 15:38:55 +08:00
|
|
|
from .auto_augment import (AutoAugment, AutoContrast, Brightness,
|
2021-04-09 14:02:50 +08:00
|
|
|
ColorTransform, Contrast, Cutout, Equalize, Invert,
|
|
|
|
Posterize, RandAugment, Rotate, Sharpness, Shear,
|
|
|
|
Solarize, SolarizeAdd, Translate)
|
2022-08-24 15:59:02 +08:00
|
|
|
from .formatting import Collect, PackClsInputs, ToNumpy, ToPIL, Transpose
|
2022-07-01 11:07:14 +08:00
|
|
|
from .processing import (Albumentations, ColorJitter, EfficientNetCenterCrop,
|
2022-08-24 15:59:02 +08:00
|
|
|
EfficientNetRandomCrop, Lighting, RandomCrop,
|
|
|
|
RandomErasing, RandomResizedCrop)
|
2020-05-21 21:21:43 +08:00
|
|
|
|
2020-07-01 16:09:06 +08:00
|
|
|
__all__ = [
|
2022-08-24 15:59:02 +08:00
|
|
|
'ToPIL', 'ToNumpy', 'Transpose', 'Collect', 'RandomCrop',
|
|
|
|
'RandomResizedCrop', 'Shear', 'Translate', 'Rotate', 'Invert',
|
2022-06-02 17:52:59 +08:00
|
|
|
'ColorTransform', 'Solarize', 'Posterize', 'AutoContrast', 'Equalize',
|
|
|
|
'Contrast', 'Brightness', 'Sharpness', 'AutoAugment', 'SolarizeAdd',
|
2022-08-24 15:59:02 +08:00
|
|
|
'Cutout', 'RandAugment', 'Lighting', 'ColorJitter', 'RandomErasing',
|
2022-07-01 11:07:14 +08:00
|
|
|
'PackClsInputs', 'Albumentations', 'EfficientNetRandomCrop',
|
|
|
|
'EfficientNetCenterCrop'
|
2020-07-01 16:09:06 +08:00
|
|
|
]
|