mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
c01abb4f30
* [Refactor] Using mmcv bricks to refactor vit * Follow the vit code structure from mmclassification * Add MMCV install into CI system. * Add to 'Install MMCV' CI item * Add 'Install MMCV_CPU' and 'Install MMCV_GPU CI' items * Fix & Add 1. Fix low code coverage of vit.py; 2. Remove HybirdEmbed; 3. Fix doc string of VisionTransformer; * Add helpers unit test. * Add converter to convert vit pretrain weights from timm style to mmcls style. * Clean some rebundant code and refactor init 1. Use timm style init_weights; 2. Remove to_xtuple and trunc_norm_; * Add comments for VisionTransformer.init_weights() * Add arg: pretrain_style to choose timm or mmcls vit pretrain weights.
14 lines
499 B
Python
14 lines
499 B
Python
from .drop import DropPath
|
|
from .inverted_residual import InvertedResidual, InvertedResidualV3
|
|
from .make_divisible import make_divisible
|
|
from .res_layer import ResLayer
|
|
from .se_layer import SELayer
|
|
from .self_attention_block import SelfAttentionBlock
|
|
from .timm_convert import vit_convert
|
|
from .up_conv_block import UpConvBlock
|
|
|
|
__all__ = [
|
|
'ResLayer', 'SelfAttentionBlock', 'make_divisible', 'InvertedResidual',
|
|
'UpConvBlock', 'InvertedResidualV3', 'SELayer', 'DropPath', 'vit_convert'
|
|
]
|