Move InvertedResidual module into mmseg/utils.

pull/58/head
johnzja 2020-08-17 13:30:42 +08:00
parent 4e7cdf927a
commit 9f128f9e11
4 changed files with 5 additions and 9 deletions

View File

@ -4,7 +4,8 @@ from mmcv.cnn import ConvModule, constant_init, kaiming_init
from torch.nn.modules.batchnorm import _BatchNorm from torch.nn.modules.batchnorm import _BatchNorm
from mmseg.models.decode_heads.psp_head import PPM from mmseg.models.decode_heads.psp_head import PPM
from mmseg.ops import DepthwiseSeparableConvModule, InvertedResidual, resize from mmseg.ops import DepthwiseSeparableConvModule, resize
from mmseg.utils import InvertedResidual
from ..builder import BACKBONES from ..builder import BACKBONES

View File

@ -1,8 +1,5 @@
from .encoding import Encoding from .encoding import Encoding
from .inverted_residual_module import InvertedResidual
from .separable_conv_module import DepthwiseSeparableConvModule from .separable_conv_module import DepthwiseSeparableConvModule
from .wrappers import resize from .wrappers import resize
__all__ = [ __all__ = ['resize', 'DepthwiseSeparableConvModule', 'Encoding']
'resize', 'DepthwiseSeparableConvModule', 'InvertedResidual', 'Encoding'
]

View File

@ -1,7 +1,5 @@
from .collect_env import collect_env from .collect_env import collect_env
from .inverted_residual_module import InvertedResidual
from .logger import get_root_logger from .logger import get_root_logger
__all__ = [ __all__ = ['get_root_logger', 'collect_env', 'InvertedResidual']
'get_root_logger',
'collect_env',
]