mirror of https://github.com/alibaba/EasyCV.git
22 lines
943 B
Python
22 lines
943 B
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
from .activation import FReLU
|
|
from .conv_module import ConvModule, build_conv_layer
|
|
from .conv_ws import ConvWS2d, conv_ws_2d
|
|
from .dist_utils import DistributedLossWrapper, DistributedMinerWrapper
|
|
from .gather_layer import GatherLayer
|
|
from .init_weights import _init_weights, trunc_normal_
|
|
from .multi_pooling import GeMPooling, MultiAvgPooling, MultiPooling
|
|
from .norm import build_norm_layer
|
|
from .pos_embed import get_2d_sincos_pos_embed, interpolate_pos_embed
|
|
from .res_layer import ResLayer
|
|
from .scale import Scale
|
|
# from .weight_init import (bias_init_with_prob, kaiming_init, normal_init,
|
|
# uniform_init, xavier_init)
|
|
from .sobel import Sobel
|
|
|
|
# __all__ = [
|
|
# 'conv_ws_2d', 'ConvWS2d', 'build_conv_layer', 'ConvModule',
|
|
# 'build_norm_layer', 'xavier_init', 'normal_init', 'uniform_init',
|
|
# 'kaiming_init', 'bias_init_with_prob', 'Scale', 'Sobel'
|
|
# ]
|