mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add BaseTransform in __init__.py (#1852) 

* Update __init__.py

* Update __init__.py
pull/2133/head
Yue Zhou 2022-04-02 16:37:31 +08:00 committed by zhouzaida
parent 428512f88a
commit 46cb4b108c
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mmcv/transforms/__init__.py
View File

@ -1,4 +1,5 @@
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from .base import BaseTransform
from .builder import TRANSFORMS from .builder import TRANSFORMS
from .loading import LoadAnnotations, LoadImageFromFile from .loading import LoadAnnotations, LoadImageFromFile
from .processing import (CenterCrop, MultiScaleFlipAug, Normalize, Pad, from .processing import (CenterCrop, MultiScaleFlipAug, Normalize, Pad,
@ -10,18 +11,18 @@ try:
import torch # noqa: F401 import torch # noqa: F401
except ImportError: except ImportError:
__all__ = [ __all__ = [
'TRANSFORMS', 'TransformBroadcaster', 'Compose', 'RandomChoice', 'BaseTransform', 'TRANSFORMS', 'TransformBroadcaster', 'Compose',
'KeyMapper', 'LoadImageFromFile', 'LoadAnnotations', 'Normalize', 'RandomChoice', 'KeyMapper', 'LoadImageFromFile', 'LoadAnnotations',
'Resize', 'Pad', 'RandomFlip', 'RandomChoiceResize', 'CenterCrop', 'Normalize', 'Resize', 'Pad', 'RandomFlip', 'RandomChoiceResize',
'RandomGrayscale', 'MultiScaleFlipAug', 'RandomResize' 'CenterCrop', 'RandomGrayscale', 'MultiScaleFlipAug', 'RandomResize'
] ]
else: else:
from .formatting import ImageToTensor, ToTensor, to_tensor from .formatting import ImageToTensor, ToTensor, to_tensor
__all__ = [ __all__ = [
'TRANSFORMS', 'TransformBroadcaster', 'Compose', 'RandomChoice', 'BaseTransform', 'TRANSFORMS', 'TransformBroadcaster', 'Compose',
'KeyMapper', 'LoadImageFromFile', 'LoadAnnotations', 'Normalize', 'RandomChoice', 'KeyMapper', 'LoadImageFromFile', 'LoadAnnotations',
'Resize', 'Pad', 'ToTensor', 'to_tensor', 'ImageToTensor', 'Normalize', 'Resize', 'Pad', 'ToTensor', 'to_tensor', 'ImageToTensor',
'RandomFlip', 'RandomChoiceResize', 'CenterCrop', 'RandomGrayscale', 'RandomFlip', 'RandomChoiceResize', 'CenterCrop', 'RandomGrayscale',
'MultiScaleFlipAug', 'RandomResize' 'MultiScaleFlipAug', 'RandomResize'
] ]