31 lines
1.7 KiB
Python
31 lines
1.7 KiB
Python
# All modification made by Kneron Corp.: Copyright (c) 2022 Kneron Corp.
|
|
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .auto_augment import (AutoAugment, BrightnessTransform, ColorTransform,
|
|
ContrastTransform, EqualizeTransform, Rotate, Shear,
|
|
Translate)
|
|
from .compose import Compose
|
|
from .formatting import (Collect, DefaultFormatBundle, ImageToTensor,
|
|
ToDataContainer, ToTensor, Transpose, to_tensor)
|
|
from .instaboost import InstaBoost
|
|
from .loading import (LoadAnnotations, LoadImageFromFile, LoadImageFromWebcam,
|
|
LoadMultiChannelImageFromFiles, LoadProposals)
|
|
from .test_time_aug import MultiScaleFlipAug
|
|
from .transforms import (Albu, CutOut, Expand, MinIoURandomCrop, MixUp, Mosaic,
|
|
Normalize, Pad, PhotoMetricDistortion, RandomAffine,
|
|
RandomCenterCropPad, RandomCrop, RandomFlip,
|
|
RandomShift, Resize, SegRescale, YOLOXHSVRandomAug,
|
|
Rotate90)
|
|
|
|
__all__ = [
|
|
'Compose', 'to_tensor', 'ToTensor', 'ImageToTensor', 'ToDataContainer',
|
|
'Transpose', 'Collect', 'DefaultFormatBundle', 'LoadAnnotations',
|
|
'LoadImageFromFile', 'LoadImageFromWebcam',
|
|
'LoadMultiChannelImageFromFiles', 'LoadProposals', 'MultiScaleFlipAug',
|
|
'Resize', 'RandomFlip', 'Pad', 'RandomCrop', 'Normalize', 'SegRescale',
|
|
'MinIoURandomCrop', 'Expand', 'PhotoMetricDistortion', 'Albu',
|
|
'InstaBoost', 'RandomCenterCropPad', 'AutoAugment', 'CutOut', 'Shear',
|
|
'Rotate', 'ColorTransform', 'EqualizeTransform', 'BrightnessTransform',
|
|
'ContrastTransform', 'Translate', 'RandomShift', 'Mosaic', 'MixUp',
|
|
'RandomAffine', 'YOLOXHSVRandomAug', 'Rotate90',
|
|
]
|