__init__.py 3.2 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
from .transforms import RandomRotation  # noqa: F401
33
from .transforms import RandomPerspective  # noqa: F401
34 35
from .transforms import Grayscale  # noqa: F401
from .transforms import ToTensor  # noqa: F401
36
from .transforms import RandomErasing  # noqa: F401
37 38 39 40 41
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
42
from .functional import affine  # noqa: F401
43
from .functional import rotate  # noqa: F401
44
from .functional import perspective  # noqa: F401
45 46 47 48 49 50 51
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
52
from .functional import erase  # noqa: F401
L
LielinJiang 已提交
53

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