mirror of https://github.com/JDAI-CV/fast-reid.git
20 lines
620 B
Python
20 lines
620 B
Python
# encoding: utf-8
|
|
"""
|
|
@author: liaoxingyu
|
|
@contact: sherlockliao01@gmail.com
|
|
"""
|
|
|
|
from .activation import *
|
|
from .batch_norm import *
|
|
from .context_block import ContextBlock
|
|
from .drop import DropPath, DropBlock2d, drop_block_2d, drop_path
|
|
from .frn import FRN, TLU
|
|
from .gather_layer import GatherLayer
|
|
from .helpers import to_ntuple, to_2tuple, to_3tuple, to_4tuple, make_divisible
|
|
from .non_local import Non_local
|
|
from .se_layer import SELayer
|
|
from .splat import SplAtConv2d, DropBlock2D
|
|
from .weight_init import (
|
|
trunc_normal_, variance_scaling_, lecun_normal_, weights_init_kaiming, weights_init_classifier
|
|
)
|