24 lines
1.3 KiB
Python
24 lines
1.3 KiB
Python
|
# Copyright (c) OpenMMLab. All rights reserved.
|
||
|
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 .formatting import (Collect, ImageToTensor, PackClsInputs, ToNumpy, ToPIL,
|
||
|
|
ToTensor, Transpose, to_tensor)
|
||
|
from .processing import (Albumentations, ColorJitter, EfficientNetCenterCrop,
|
||
|
|
EfficientNetRandomCrop, Lighting, Normalize, Pad,
|
||
|
|
RandomCrop, RandomErasing, RandomGrayscale,
|
||
|
|
RandomResizedCrop)
|
||
|
|
|
||
|
__all__ = [
|
||
|
|
'Compose', 'to_tensor', 'ToTensor', 'ImageToTensor', 'ToPIL', 'ToNumpy',
|
||
|
|
'Transpose', 'Collect', 'Normalize', 'RandomCrop', 'RandomResizedCrop',
|
||
|
|
'RandomGrayscale', 'Shear', 'Translate', 'Rotate', 'Invert',
|
||
|
|
'ColorTransform', 'Solarize', 'Posterize', 'AutoContrast', 'Equalize',
|
||
|
|
'Contrast', 'Brightness', 'Sharpness', 'AutoAugment', 'SolarizeAdd',
|
||
|
|
'Cutout', 'RandAugment', 'Lighting', 'ColorJitter', 'RandomErasing', 'Pad',
|
||
|
|
'PackClsInputs', 'Albumentations', 'EfficientNetRandomCrop',
|
||
|
|
'EfficientNetCenterCrop'
|
||
|
|
]
|