| 1234567891011121314151617181920212223 |
- 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
|
