vision
/
hubconf.py
85 строк · 2.5 Кб
1# Optional list of dependencies required by the package
2dependencies = ["torch"]
3
4from torchvision.models import get_model_weights, get_weight
5from torchvision.models.alexnet import alexnet
6from torchvision.models.convnext import convnext_base, convnext_large, convnext_small, convnext_tiny
7from torchvision.models.densenet import densenet121, densenet161, densenet169, densenet201
8from torchvision.models.efficientnet import (
9efficientnet_b0,
10efficientnet_b1,
11efficientnet_b2,
12efficientnet_b3,
13efficientnet_b4,
14efficientnet_b5,
15efficientnet_b6,
16efficientnet_b7,
17efficientnet_v2_l,
18efficientnet_v2_m,
19efficientnet_v2_s,
20)
21from torchvision.models.googlenet import googlenet
22from torchvision.models.inception import inception_v3
23from torchvision.models.maxvit import maxvit_t
24from torchvision.models.mnasnet import mnasnet0_5, mnasnet0_75, mnasnet1_0, mnasnet1_3
25from torchvision.models.mobilenetv2 import mobilenet_v2
26from torchvision.models.mobilenetv3 import mobilenet_v3_large, mobilenet_v3_small
27from torchvision.models.optical_flow import raft_large, raft_small
28from torchvision.models.regnet import (
29regnet_x_16gf,
30regnet_x_1_6gf,
31regnet_x_32gf,
32regnet_x_3_2gf,
33regnet_x_400mf,
34regnet_x_800mf,
35regnet_x_8gf,
36regnet_y_128gf,
37regnet_y_16gf,
38regnet_y_1_6gf,
39regnet_y_32gf,
40regnet_y_3_2gf,
41regnet_y_400mf,
42regnet_y_800mf,
43regnet_y_8gf,
44)
45from torchvision.models.resnet import (
46resnet101,
47resnet152,
48resnet18,
49resnet34,
50resnet50,
51resnext101_32x8d,
52resnext101_64x4d,
53resnext50_32x4d,
54wide_resnet101_2,
55wide_resnet50_2,
56)
57from torchvision.models.segmentation import (
58deeplabv3_mobilenet_v3_large,
59deeplabv3_resnet101,
60deeplabv3_resnet50,
61fcn_resnet101,
62fcn_resnet50,
63lraspp_mobilenet_v3_large,
64)
65from torchvision.models.shufflenetv2 import (
66shufflenet_v2_x0_5,
67shufflenet_v2_x1_0,
68shufflenet_v2_x1_5,
69shufflenet_v2_x2_0,
70)
71from torchvision.models.squeezenet import squeezenet1_0, squeezenet1_1
72from torchvision.models.swin_transformer import swin_b, swin_s, swin_t, swin_v2_b, swin_v2_s, swin_v2_t
73from torchvision.models.vgg import vgg11, vgg11_bn, vgg13, vgg13_bn, vgg16, vgg16_bn, vgg19, vgg19_bn
74from torchvision.models.video import (
75mc3_18,
76mvit_v1_b,
77mvit_v2_s,
78r2plus1d_18,
79r3d_18,
80s3d,
81swin3d_b,
82swin3d_s,
83swin3d_t,
84)
85from torchvision.models.vision_transformer import vit_b_16, vit_b_32, vit_h_14, vit_l_16, vit_l_32
86