Skip to content

GitLab

Commit 46cb4b10 authored by Yue Zhou's avatar Yue Zhou Committed by zhouzaida
Browse files

Add BaseTransform in __init__.py (#1852) 

* Update __init__.py

* Update __init__.py
parent 428512f8
Show whitespace changes
Inline Side-by-side
Showing with 8 additions and 7 deletions
mmcv/transforms/__init__.py
View file @ 46cb4b10
# 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: ...@@ -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'
] ]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment