23 lines
865 B
Python
23 lines
865 B
Python
from .alexnet import *
|
|
from .convnext import *
|
|
from .densenet import *
|
|
from .efficientnet import *
|
|
from .googlenet import *
|
|
from .inception import *
|
|
from .mnasnet import *
|
|
from .mobilenet import *
|
|
from .regnet import *
|
|
from .resnet import *
|
|
from .shufflenetv2 import *
|
|
from .squeezenet import *
|
|
from .vgg import *
|
|
from .vision_transformer import *
|
|
from .swin_transformer import *
|
|
from .maxvit import *
|
|
from . import detection, optical_flow, quantization, segmentation, video
|
|
|
|
# The Weights and WeightsEnum are developer-facing utils that we make public for
|
|
# downstream libs like torchgeo https://github.com/pytorch/vision/issues/7094
|
|
# TODO: we could / should document them publicly, but it's not clear where, as
|
|
# they're not intended for end users.
|
|
from ._api import get_model, get_model_builder, get_model_weights, get_weight, list_models, Weights, WeightsEnum
|