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
Hide 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.
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'
]
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