mirror of
https://github.com/deepinsight/insightface.git
synced 2026-05-19 15:41:33 +00:00
28 lines
1.5 KiB
Python
Executable File
28 lines
1.5 KiB
Python
Executable File
from .auto_augment import (AutoAugment, BrightnessTransform, ColorTransform,
|
|
ContrastTransform, EqualizeTransform, Rotate, Shear,
|
|
Translate)
|
|
from .compose import Compose
|
|
from .formating 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, Normalize,
|
|
Pad, PhotoMetricDistortion, RandomCenterCropPad,
|
|
RandomSquareCrop,
|
|
RandomCrop, RandomFlip, Resize, SegRescale)
|
|
|
|
__all__ = [
|
|
'Compose', 'to_tensor', 'ToTensor', 'ImageToTensor', 'ToDataContainer',
|
|
'Transpose', 'Collect', 'DefaultFormatBundle', 'LoadAnnotations',
|
|
'LoadImageFromFile', 'LoadImageFromWebcam',
|
|
'LoadMultiChannelImageFromFiles', 'LoadProposals', 'MultiScaleFlipAug',
|
|
'Resize', 'RandomFlip', 'Pad', 'RandomCrop', 'Normalize', 'SegRescale',
|
|
'RandomSquareCrop',
|
|
'MinIoURandomCrop', 'Expand', 'PhotoMetricDistortion', 'Albu',
|
|
'InstaBoost', 'RandomCenterCropPad', 'AutoAugment', 'CutOut', 'Shear',
|
|
'Rotate', 'ColorTransform', 'EqualizeTransform', 'BrightnessTransform',
|
|
'ContrastTransform', 'Translate'
|
|
]
|