-
Notifications
You must be signed in to change notification settings - Fork 18
/
Copy path__init__.py
22 lines (22 loc) · 2.38 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from .convnext import convnext_tiny, convnext_small, convnext_base, convnext_large, convnext_xlarge
from .darknet import darknet53, darknet19, csp_darknet53, csp_darknet19
from .densenet import densenet121, densenet169, densenet201
from .effnet import effnet_b0, effnet_b1, effnet_b2, effnet_b3, effnet_b4, effnet_b5, effnet_b6, effnet_b7
from .effnet import effnet_lite_b0, effnet_lite_b1, effnet_lite_b2, effnet_lite_b3, effnet_lite_b4
from .effnet import effnet_v2_b0, effnet_v2_b1, effnet_v2_b2, effnet_v2_b3, effnet_v2_s, effnet_v2_m, effnet_v2_l
from .mobilenet import mobilenet_v2, mobilenet_v3_small, mobilenet_v3_large
from .resnet import resnet18, resnet34, resnet50, resnet101, resnet152, resnext50_32x4d, resnext101_32x8d, resnext101_64x4d, wide_resnet50_2, wide_resnet101_2
from .resnest import resnest50, resnest101, resnest200, resnest269
from .vgg import vgg11, vgg11_bn, vgg13, vgg13_bn, vgg16, vgg16_bn, vgg19, vgg19_bn
#from .swin_transformer import (swin_transformer_tiny_224_w7_1k, swin_transformer_tiny_224_w7_22k, swin_transformer_tiny_224_w7_22kto1k,
# swin_transformer_small_224_w7_1k, swin_transformer_small_224_w7_22k, swin_transformer_small_224_w7_22kto1k,
# swin_transformer_base_224_w7_1k, swin_transformer_base_224_w7_22k, swin_transformer_base_224_w7_22kto1k,
# swin_transformer_base_384_w12_1k, swin_transformer_base_384_w12_22k, swin_transformer_base_384_w12_22kto1k,
# swin_transformer_large_224_w7_22k, swin_transformer_large_224_w7_22kto1k,
# swin_transformer_large_384_w12_22k, swin_transformer_large_384_w12_22kto1k)
#from .swin_transformer import (swin_transformer_v2_tiny_256_w8_1k, swin_transformer_v2_tiny_256_w16_1k,
# swin_transformer_v2_small_256_w8_1k, swin_transformer_v2_small_256_w16_1k,
# swin_transformer_v2_base_256_w8_1k, swin_transformer_v2_base_256_w16_1k,
# swin_transformer_v2_base_192_w12_22k, swin_transformer_v2_base_256_w16_22kto1k, swin_transformer_v2_base_384_w24_22kto1k,
# swin_transformer_v2_large_192_w12_22k, swin_transformer_v2_large_256_w16_22kto1k, swin_transformer_v2_large_384_w24_22kto1k)
from .swin_transformer import swin_tiny, swin_small, swin_base, swin_v2_tiny, swin_v2_small, swin_v2_base