__init__.py 3.0 KB
Newer Older
L
LielinJiang 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14
# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
from .transforms import BaseTransform  # noqa: F401
from .transforms import Compose  # noqa: F401
from .transforms import Resize  # noqa: F401
from .transforms import RandomResizedCrop  # noqa: F401
from .transforms import CenterCrop  # noqa: F401
from .transforms import RandomHorizontalFlip  # noqa: F401
from .transforms import RandomVerticalFlip  # noqa: F401
from .transforms import Transpose  # noqa: F401
from .transforms import Normalize  # noqa: F401
from .transforms import BrightnessTransform  # noqa: F401
from .transforms import SaturationTransform  # noqa: F401
from .transforms import ContrastTransform  # noqa: F401
from .transforms import HueTransform  # noqa: F401
from .transforms import ColorJitter  # noqa: F401
from .transforms import RandomCrop  # noqa: F401
from .transforms import Pad  # noqa: F401
31
from .transforms import RandomAffine  # noqa: F401
32 33 34
from .transforms import RandomRotation  # noqa: F401
from .transforms import Grayscale  # noqa: F401
from .transforms import ToTensor  # noqa: F401
35
from .transforms import RandomErasing  # noqa: F401
36 37 38 39 40
from .functional import to_tensor  # noqa: F401
from .functional import hflip  # noqa: F401
from .functional import vflip  # noqa: F401
from .functional import resize  # noqa: F401
from .functional import pad  # noqa: F401
41
from .functional import affine  # noqa: F401
42 43 44 45 46 47 48 49
from .functional import rotate  # noqa: F401
from .functional import to_grayscale  # noqa: F401
from .functional import crop  # noqa: F401
from .functional import center_crop  # noqa: F401
from .functional import adjust_brightness  # noqa: F401
from .functional import adjust_contrast  # noqa: F401
from .functional import adjust_hue  # noqa: F401
from .functional import normalize  # noqa: F401
50
from .functional import erase  # noqa: F401
L
LielinJiang 已提交
51

52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68
__all__ = [ #noqa
    'BaseTransform',
    'Compose',
    'Resize',
    'RandomResizedCrop',
    'CenterCrop',
    'RandomHorizontalFlip',
    'RandomVerticalFlip',
    'Transpose',
    'Normalize',
    'BrightnessTransform',
    'SaturationTransform',
    'ContrastTransform',
    'HueTransform',
    'ColorJitter',
    'RandomCrop',
    'Pad',
69
    'RandomAffine',
70 71 72
    'RandomRotation',
    'Grayscale',
    'ToTensor',
73
    'RandomErasing',
74 75 76 77 78
    'to_tensor',
    'hflip',
    'vflip',
    'resize',
    'pad',
79
    'affine',
80 81 82 83 84 85 86
    'rotate',
    'to_grayscale',
    'crop',
    'center_crop',
    'adjust_brightness',
    'adjust_contrast',
    'adjust_hue',
87 88
    'normalize',
    'erase',
89
]