98 lines
2.0 KiB
Python
98 lines
2.0 KiB
Python
# 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.
|
|
|
|
from .functional import (
|
|
adjust_brightness,
|
|
adjust_contrast,
|
|
adjust_hue,
|
|
affine,
|
|
center_crop,
|
|
crop,
|
|
erase,
|
|
hflip,
|
|
normalize,
|
|
pad,
|
|
perspective,
|
|
resize,
|
|
rotate,
|
|
to_grayscale,
|
|
to_tensor,
|
|
vflip,
|
|
)
|
|
from .transforms import (
|
|
BaseTransform,
|
|
BrightnessTransform,
|
|
CenterCrop,
|
|
ColorJitter,
|
|
Compose,
|
|
ContrastTransform,
|
|
Grayscale,
|
|
HueTransform,
|
|
Normalize,
|
|
Pad,
|
|
RandomAffine,
|
|
RandomCrop,
|
|
RandomErasing,
|
|
RandomHorizontalFlip,
|
|
RandomPerspective,
|
|
RandomResizedCrop,
|
|
RandomRotation,
|
|
RandomVerticalFlip,
|
|
Resize,
|
|
SaturationTransform,
|
|
ToTensor,
|
|
Transpose,
|
|
)
|
|
|
|
__all__ = [
|
|
'BaseTransform',
|
|
'Compose',
|
|
'Resize',
|
|
'RandomResizedCrop',
|
|
'CenterCrop',
|
|
'RandomHorizontalFlip',
|
|
'RandomVerticalFlip',
|
|
'Transpose',
|
|
'Normalize',
|
|
'BrightnessTransform',
|
|
'SaturationTransform',
|
|
'ContrastTransform',
|
|
'HueTransform',
|
|
'ColorJitter',
|
|
'RandomCrop',
|
|
'Pad',
|
|
'RandomAffine',
|
|
'RandomRotation',
|
|
'RandomPerspective',
|
|
'Grayscale',
|
|
'ToTensor',
|
|
'RandomErasing',
|
|
'to_tensor',
|
|
'hflip',
|
|
'vflip',
|
|
'resize',
|
|
'pad',
|
|
'affine',
|
|
'rotate',
|
|
'perspective',
|
|
'to_grayscale',
|
|
'crop',
|
|
'center_crop',
|
|
'adjust_brightness',
|
|
'adjust_contrast',
|
|
'adjust_hue',
|
|
'normalize',
|
|
'erase',
|
|
]
|