42 lines
2.4 KiB
Python
42 lines
2.4 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .alexnet import AlexNet
|
|
# yapf: disable
|
|
from .bricks import (ACTIVATION_LAYERS, CONV_LAYERS, NORM_LAYERS,
|
|
PADDING_LAYERS, PLUGIN_LAYERS, UPSAMPLE_LAYERS,
|
|
ContextBlock, Conv2d, Conv3d, ConvAWS2d, ConvModule,
|
|
ConvTranspose2d, ConvTranspose3d, ConvWS2d,
|
|
DepthwiseSeparableConvModule, GeneralizedAttention,
|
|
HSigmoid, HSwish, Linear, MaxPool2d, MaxPool3d,
|
|
NonLocal1d, NonLocal2d, NonLocal3d, Scale, Swish,
|
|
build_activation_layer, build_conv_layer,
|
|
build_norm_layer, build_padding_layer, build_plugin_layer,
|
|
build_upsample_layer, conv_ws_2d, is_norm)
|
|
from .builder import MODELS, build_model_from_cfg
|
|
# yapf: enable
|
|
from .resnet import ResNet, make_res_layer
|
|
from .utils import (INITIALIZERS, Caffe2XavierInit, ConstantInit, KaimingInit,
|
|
NormalInit, PretrainedInit, TruncNormalInit, UniformInit,
|
|
XavierInit, bias_init_with_prob, caffe2_xavier_init,
|
|
constant_init, fuse_conv_bn, get_model_complexity_info,
|
|
initialize, kaiming_init, normal_init, trunc_normal_init,
|
|
uniform_init, xavier_init)
|
|
from .vgg import VGG, make_vgg_layer
|
|
|
|
__all__ = [
|
|
'AlexNet', 'VGG', 'make_vgg_layer', 'ResNet', 'make_res_layer',
|
|
'constant_init', 'xavier_init', 'normal_init', 'trunc_normal_init',
|
|
'uniform_init', 'kaiming_init', 'caffe2_xavier_init',
|
|
'bias_init_with_prob', 'ConvModule', 'build_activation_layer',
|
|
'build_conv_layer', 'build_norm_layer', 'build_padding_layer',
|
|
'build_upsample_layer', 'build_plugin_layer', 'is_norm', 'NonLocal1d',
|
|
'NonLocal2d', 'NonLocal3d', 'ContextBlock', 'HSigmoid', 'Swish', 'HSwish',
|
|
'GeneralizedAttention', 'ACTIVATION_LAYERS', 'CONV_LAYERS', 'NORM_LAYERS',
|
|
'PADDING_LAYERS', 'UPSAMPLE_LAYERS', 'PLUGIN_LAYERS', 'Scale',
|
|
'get_model_complexity_info', 'conv_ws_2d', 'ConvAWS2d', 'ConvWS2d',
|
|
'fuse_conv_bn', 'DepthwiseSeparableConvModule', 'Linear', 'Conv2d',
|
|
'ConvTranspose2d', 'MaxPool2d', 'ConvTranspose3d', 'MaxPool3d', 'Conv3d',
|
|
'initialize', 'INITIALIZERS', 'ConstantInit', 'XavierInit', 'NormalInit',
|
|
'TruncNormalInit', 'UniformInit', 'KaimingInit', 'PretrainedInit',
|
|
'Caffe2XavierInit', 'MODELS', 'build_model_from_cfg'
|
|
]
|