mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
* select_conv2d -> create_conv2d * added create_attn to create attention module from string/bool/module * factor padding helpers into own file, use in both conv2d_same and avg_pool2d_same * add some more test eca resnet variants * minor tweaks, naming, comments, consistency
18 lines
773 B
Python
18 lines
773 B
Python
from .padding import get_padding
|
|
from .avg_pool2d_same import AvgPool2dSame
|
|
from .conv2d_same import Conv2dSame
|
|
from .conv_bn_act import ConvBnAct
|
|
from .mixed_conv2d import MixedConv2d
|
|
from .cond_conv2d import CondConv2d, get_condconv_initializer
|
|
from .create_conv2d import create_conv2d
|
|
from .create_attn import create_attn
|
|
from .selective_kernel import SelectiveKernelConv
|
|
from .se import SEModule
|
|
from .eca import EcaModule, CecaModule
|
|
from .activations import *
|
|
from .adaptive_avgmax_pool import \
|
|
adaptive_avgmax_pool2d, select_adaptive_pool2d, AdaptiveAvgMaxPool2d, SelectAdaptivePool2d
|
|
from .drop import DropBlock2d, DropPath
|
|
from .test_time_pool import TestTimePoolHead, apply_test_time_pool
|
|
from .split_batchnorm import SplitBatchNorm2d, convert_splitbn_model
|