EasyCV/easycv/models/utils/__init__.py

# 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,
                         get_world_size, is_dist_avail_and_initialized,
                         reduce_mean)
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
from .transformer import (MLP, TransformerEncoder, TransformerEncoderLayer,
                          _get_activation_fn, _get_clones)

# __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'
# ]