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.
from .base import BaseTransform
from .builder import TRANSFORMS
from .loading import LoadAnnotations, LoadImageFromFile
from .processing import (CenterCrop, MultiScaleFlipAug, Normalize, Pad,
@ -10,18 +11,18 @@ try:
import torch # noqa: F401
except ImportError:
__all__ = [
'TRANSFORMS', 'TransformBroadcaster', 'Compose', 'RandomChoice',
'KeyMapper', 'LoadImageFromFile', 'LoadAnnotations', 'Normalize',
'Resize', 'Pad', 'RandomFlip', 'RandomChoiceResize', 'CenterCrop',
'RandomGrayscale', 'MultiScaleFlipAug', 'RandomResize'
'BaseTransform', 'TRANSFORMS', 'TransformBroadcaster', 'Compose',
'RandomChoice', 'KeyMapper', 'LoadImageFromFile', 'LoadAnnotations',
'Normalize', 'Resize', 'Pad', 'RandomFlip', 'RandomChoiceResize',
'CenterCrop', 'RandomGrayscale', 'MultiScaleFlipAug', 'RandomResize'
]
else:
from .formatting import ImageToTensor, ToTensor, to_tensor
__all__ = [
'TRANSFORMS', 'TransformBroadcaster', 'Compose', 'RandomChoice',
'KeyMapper', 'LoadImageFromFile', 'LoadAnnotations', 'Normalize',
'Resize', 'Pad', 'ToTensor', 'to_tensor', 'ImageToTensor',
'BaseTransform', 'TRANSFORMS', 'TransformBroadcaster', 'Compose',
'RandomChoice', 'KeyMapper', 'LoadImageFromFile', 'LoadAnnotations',
'Normalize', 'Resize', 'Pad', 'ToTensor', 'to_tensor', 'ImageToTensor',
'RandomFlip', 'RandomChoiceResize', 'CenterCrop', 'RandomGrayscale',
'MultiScaleFlipAug', 'RandomResize'
]