mirror of
https://github.com/alibaba/EasyCV.git
synced 2025-06-03 14:49:00 +08:00
30 lines
1.4 KiB
Python
30 lines
1.4 KiB
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,
|
|
reduce_mean)
|
|
from .face_keypoint_utils import (ION, InvertedResidual, Residual, Softmax,
|
|
View, conv_bn, conv_no_relu,
|
|
get_keypoint_accuracy, get_pose_accuracy,
|
|
pose_accuracy)
|
|
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, ConvMlp, DropPath, 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'
|
|
# ]
|