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 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
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -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'
|
|
||||||
]
|
|
||||||
|
|
|
@ -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',
|
|
||||||
]
|
|
||||||
|
|
Loading…
Reference in New Issue