OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
31 lines
1.8 KiB
31 lines
1.8 KiB
# 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 (FilterAnnotations, LoadAnnotations, LoadImageFromFile, |
|
LoadImageFromWebcam, LoadMultiChannelImageFromFiles, |
|
LoadPanopticAnnotations, LoadProposals) |
|
from .test_time_aug import MultiScaleFlipAug |
|
from .transforms import (Albu, CopyPaste, CutOut, Expand, MinIoURandomCrop, |
|
MixUp, Mosaic, Normalize, Pad, PhotoMetricDistortion, |
|
RandomAffine, RandomCenterCropPad, RandomCrop, |
|
RandomFlip, RandomShift, Resize, SegRescale, |
|
YOLOXHSVRandomAug) |
|
|
|
__all__ = [ |
|
'Compose', 'to_tensor', 'ToTensor', 'ImageToTensor', 'ToDataContainer', |
|
'Transpose', 'Collect', 'DefaultFormatBundle', 'LoadAnnotations', |
|
'LoadImageFromFile', 'LoadImageFromWebcam', 'LoadPanopticAnnotations', |
|
'LoadMultiChannelImageFromFiles', 'LoadProposals', 'FilterAnnotations', |
|
'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', 'CopyPaste' |
|
]
|
|
|