Move InvertedResidual module into mmseg/utils.
parent
4e7cdf927a
commit
9f128f9e11
|
@ -4,7 +4,8 @@ from mmcv.cnn import ConvModule, constant_init, kaiming_init
|
|||
from torch.nn.modules.batchnorm import _BatchNorm
|
||||
|
||||
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
|
||||
|
||||
|
||||
|
|
|
@ -1,8 +1,5 @@
|
|||
from .encoding import Encoding
|
||||
from .inverted_residual_module import InvertedResidual
|
||||
from .separable_conv_module import DepthwiseSeparableConvModule
|
||||
from .wrappers import resize
|
||||
|
||||
__all__ = [
|
||||
'resize', 'DepthwiseSeparableConvModule', 'InvertedResidual', 'Encoding'
|
||||
]
|
||||
__all__ = ['resize', 'DepthwiseSeparableConvModule', 'Encoding']
|
||||
|
|
|
@ -1,7 +1,5 @@
|
|||
from .collect_env import collect_env
|
||||
from .inverted_residual_module import InvertedResidual
|
||||
from .logger import get_root_logger
|
||||
|
||||
__all__ = [
|
||||
'get_root_logger',
|
||||
'collect_env',
|
||||
]
|
||||
__all__ = ['get_root_logger', 'collect_env', 'InvertedResidual']
|
||||
|
|
Loading…
Reference in New Issue