flowvision.models.rst

flowvision.models

Pretrain Models for Visual Tasks

Classification

The models subpackage contains definitions for the following model architectures for image classification:

• AlexNet
• SqueezeNet
• VGG
• GoogLeNet
• InceptionV3
• ResNet
• ResNeXt
• ResNeSt
• SENet
• DenseNet
• ShuffleNetV2
• MobileNetV2
• MobileNetV3
• MNASNet
• GhostNet
• Res2Net
• EfficientNet
• RegNet
• ReXNet
• ViT
• DeiT
• PVT
• Swin-Transformer
• CSwin-Transformer
• CrossFormer
• PoolFormer
• Mlp_Mixer
• ResMLP
• gMLP
• ConvMixer
• ConvNeXt
• LeViT
• RegionViT
• VAN
• MobileViT
• DeiT-III
• CaiT
• DLA
• GENet
• HRNet
• FAN

.. currentmodule:: flowvision.models

Alexnet

.. automodule:: flowvision.models
    :members: alexnet

SqueezeNet

.. automodule:: flowvision.models
    :members:
        squeezenet1_0,
        squeezenet1_1,

VGG

.. automodule:: flowvision.models
    :members:
        vgg11,
        vgg11_bn,
        vgg13,
        vgg13_bn,
        vgg16,
        vgg16_bn,
        vgg19,
        vgg19_bn,

GoogLeNet

.. automodule:: flowvision.models
    :members:
        googlenet,

InceptionV3

.. automodule:: flowvision.models
    :members:
        inception_v3,

ResNet

.. automodule:: flowvision.models
    :members:
        resnet18,
        resnet34,
        resnet50,
        resnet101,
        resnet152,
        resnext50_32x4d,
        resnext101_32x8d,
        wide_resnet50_2,
        wide_resnet101_2,

ResNeSt

.. automodule:: flowvision.models
    :members:
        resnest50,
        resnest101,
        resnest200,
        resnest269,

DenseNet

.. automodule:: flowvision.models
    :members:
        densenet121,
        densenet169,
        densenet201,
        densenet161,

ShuffleNetV2

.. automodule:: flowvision.models
    :members:
        shufflenet_v2_x0_5,
        shufflenet_v2_x1_0,
        shufflenet_v2_x1_5,
        shufflenet_v2_x2_0,

MobileNetV2

.. automodule:: flowvision.models
    :members:
        mobilenet_v2

MobileNetV3

.. automodule:: flowvision.models
    :members:
        mobilenet_v3_small,
        mobilenet_v3_large,

MNASNet

.. automodule:: flowvision.models
    :members:
        mnasnet0_5,
        mnasnet0_75,
        mnasnet1_0,
        mnasnet1_3,

GhostNet

.. automodule:: flowvision.models
    :members:
        ghostnet,

Res2Net

.. automodule:: flowvision.models
    :members:
        res2net50_26w_4s,
        res2net50_26w_6s,
        res2net50_26w_8s,
        res2net50_48w_2s,
        res2net50_14w_8s,
        res2net101_26w_4s,

EfficientNet

.. automodule:: flowvision.models
    :members:
        efficientnet_b0,
        efficientnet_b1,
        efficientnet_b2,
        efficientnet_b3,
        efficientnet_b4,
        efficientnet_b5,
        efficientnet_b6,
        efficientnet_b7

RegNet

.. automodule:: flowvision.models
    :members:
        regnet_y_400mf,
        regnet_y_800mf,
        regnet_y_1_6gf,
        regnet_y_3_2gf,
        regnet_y_8gf,
        regnet_y_16gf,
        regnet_y_32gf,
        regnet_x_400mf,
        regnet_x_800mf,
        regnet_x_1_6gf,
        regnet_x_3_2gf,
        regnet_x_8gf,
        regnet_x_16gf,
        regnet_x_32gf,

ReXNet

.. automodule:: flowvision.models
    :members:
        rexnetv1_1_0,
        rexnetv1_1_3,
        rexnetv1_1_5,
        rexnetv1_2_0,
        rexnetv1_3_0,

.. automodule:: flowvision.models
    :members:
        rexnet_lite_1_0,
        rexnet_lite_1_3,
        rexnet_lite_1_5,
        rexnet_lite_2_0,

SENet

.. automodule:: flowvision.models
    :members:
        senet154,
        se_resnet50,
        se_resnet101,
        se_resnet152,
        se_resnext50_32x4d,
        se_resnext101_32x4d,

ViT

.. automodule:: flowvision.models
    :members:
        vit_tiny_patch16_224,
        vit_tiny_patch16_384,
        vit_small_patch32_224,
        vit_small_patch32_384,
        vit_small_patch16_224,
        vit_small_patch16_384,
        vit_base_patch32_224,
        vit_base_patch32_384,
        vit_base_patch16_224,
        vit_base_patch16_384,
        vit_base_patch8_224,
        vit_large_patch32_224,
        vit_large_patch32_384,
        vit_large_patch16_224,
        vit_large_patch16_384,
        vit_base_patch16_224_sam,
        vit_base_patch32_224_sam,
        vit_huge_patch14_224,
        vit_giant_patch14_224,
        vit_gigantic_patch14_224,
        vit_tiny_patch16_224_in21k,
        vit_small_patch32_224_in21k,
        vit_small_patch16_224_in21k,
        vit_base_patch32_224_in21k,
        vit_base_patch16_224_in21k,
        vit_base_patch8_224_in21k,
        vit_large_patch32_224_in21k,
        vit_large_patch16_224_in21k,
        vit_huge_patch14_224_in21k,
        vit_base_patch16_224_miil_in21k,
        vit_base_patch16_224_miil,

DeiT

.. automodule:: flowvision.models
    :members:
        deit_tiny_patch16_224,
        deit_small_patch16_224,
        deit_base_patch16_224,
        deit_base_patch16_384,
        deit_tiny_distilled_patch16_224,
        deit_small_distilled_patch16_224,
        deit_base_distilled_patch16_224,
        deit_base_distilled_patch16_384,

PVT

.. automodule:: flowvision.models
    :members:
        pvt_tiny,
        pvt_small,
        pvt_medium,
        pvt_large,

Swin-Transformer

.. automodule:: flowvision.models
    :members:
        swin_tiny_patch4_window7_224,
        swin_small_patch4_window7_224,
        swin_base_patch4_window7_224,
        swin_base_patch4_window12_384,
        swin_base_patch4_window7_224_in22k_to_1k,
        swin_base_patch4_window12_384_in22k_to_1k,
        swin_large_patch4_window7_224_in22k_to_1k,
        swin_large_patch4_window12_384_in22k_to_1k,

CSwin-Transformer

.. automodule:: flowvision.models
    :members:
        cswin_tiny_224,
        cswin_small_224,
        cswin_base_224,
        cswin_large_224,
        cswin_base_384,
        cswin_large_384,

CrossFormer

.. automodule:: flowvision.models
    :members:
        crossformer_tiny_patch4_group7_224,
        crossformer_small_patch4_group7_224,
        crossformer_base_patch4_group7_224,
        crossformer_large_patch4_group7_224,

PoolFormer

.. automodule:: flowvision.models
    :members:
        poolformer_s12,
        poolformer_s24,
        poolformer_s36,
        poolformer_m36,
        poolformer_m48,

UniFormer

.. automodule:: flowvision.models
    :members:
        uniformer_small,
        uniformer_small_plus,
        uniformer_base,
        uniformer_base_ls,

Mlp-Mixer

.. automodule:: flowvision.models
    :members:
        mlp_mixer_s16_224,
        mlp_mixer_s32_224,
        mlp_mixer_b16_224,
        mlp_mixer_b32_224,
        mlp_mixer_b16_224_in21k,
        mlp_mixer_l16_224,
        mlp_mixer_l32_224,
        mlp_mixer_l16_224_in21k,
        mlp_mixer_b16_224_miil,
        mlp_mixer_b16_224_miil_in21k,

ResMLP

.. automodule:: flowvision.models
    :members:
        resmlp_12_224,
        resmlp_12_distilled_224,
        resmlp_12_224_dino,
        resmlp_24_224,
        resmlp_24_distilled_224,
        resmlp_24_224_dino,
        resmlp_36_224,
        resmlp_36_distilled_224,
        resmlp_big_24_224,
        resmlp_big_24_224_in22k_to_1k,
        resmlp_big_24_distilled_224,

gMLP

.. automodule:: flowvision.models
    :members:
        gmlp_ti16_224,
        gmlp_s16_224,
        gmlp_b16_224,

ConvMixer

.. automodule:: flowvision.models
    :members:
        convmixer_1536_20,
        convmixer_768_32_relu,
        convmixer_1024_20,

ConvNeXt

.. automodule:: flowvision.models
    :members:
        convnext_tiny_224,
        convnext_small_224,
        convnext_base_224,
        convnext_base_384,
        convnext_large_224,
        convnext_large_384,
        convnext_base_224_22k,
        convnext_base_224_22k_to_1k,
        convnext_base_384_22k_to_1k,
        convnext_large_224_22k,
        convnext_large_224_22k_to_1k,
        convnext_large_384_22k_to_1k,
        convnext_xlarge_224_22k,
        convnext_xlarge_224_22k_to_1k,
        convnext_xlarge_384_22k_to_1k,
        convnext_iso_small_224,
        convnext_iso_base_224,
        convnext_iso_large_224,

RegionViT

.. automodule:: flowvision.models
    :members:
        regionvit_tiny_224,
        regionvit_small_224,
        regionvit_small_w14_224,
        regionvit_small_w14_peg_224,
        regionvit_medium_224,
        regionvit_base_224,
        regionvit_base_w14_224,
        regionvit_base_w14_peg_224,

VAN

.. automodule:: flowvision.models
    :members:
        van_tiny,
        van_small,
        van_base,
        van_large,

LeViT

.. automodule:: flowvision.models
    :members:
        levit_128s,
        levit_128,
        levit_192,
        levit_256,
        levit_384,

MobileViT

.. automodule:: flowvision.models
    :members:
        mobilevit_small,
        mobilevit_x_small,
        mobilevit_xx_small,

DeiT-III

.. automodule:: flowvision.models
    :members:
        deit_small_patch16_LS_224,
        deit_small_patch16_LS_384,
        deit_small_patch16_LS_224_in21k,
        deit_small_patch16_LS_384_in21k,
        deit_base_patch16_LS_224,
        deit_base_patch16_LS_384,
        deit_base_patch16_LS_224_in21k,
        deit_base_patch16_LS_384_in21k,
        deit_large_patch16_LS_224,
        deit_large_patch16_LS_384,
        deit_large_patch16_LS_224_in21k,
        deit_large_patch16_LS_384_in21k,
        deit_huge_patch14_LS_224,
        deit_huge_patch14_LS_224_in21k,

CaiT

.. automodule:: flowvision.models
    :members:
        cait_XS24_384,
        cait_S24_224,
        cait_S24_384,
        cait_S36_384,
        cait_M36_384,
        cait_M48_448,

DLA

.. automodule:: flowvision.models
    :members:
        dla34,
        dla46_c,
        dla46x_c,
        dla60x_c,
        dla60,
        dla60x,
        dla102,
        dla102x,
        dla102x2,
        dla169,

GENet

.. automodule:: flowvision.models
    :members:
        genet_small,
        genet_normal,
        genet_large

HRNet

.. automodule:: flowvision.models
    :members:
        hrnet_w18_small,
        hrnet_w18_small_v2,
        hrnet_w18,
        hrnet_w30,
        hrnet_w32,
        hrnet_w40,
        hrnet_w44,
        hrnet_w48,
        hrnet_w64

FAN

.. automodule:: flowvision.models
    :members:
        fan_tiny_12_p16_224,
        fan_small_12_p16_224,
        fan_base_18_p16_224,
        fan_large_24_p16_224,
        fan_tiny_8_p4_hybrid,
        fan_small_12_p4_hybrid,
        fan_base_16_p4_hybrid,
        fan_large_16_p4_hybrid,
        fan_base_16_p4_hybrid_in22k_1k,
        fan_base_16_p4_hybrid_in22k_1k_384,
        fan_large_16_p4_hybrid_in22k_1k,
        fan_large_16_p4_hybrid_in22k_1k_384

Neural Style Transfer

.. currentmodule:: flowvision.models

.. automodule:: flowvision.models.style_transfer
    :members:
        fast_neural_style,

Face Recognition

.. currentmodule:: flowvision.models

.. automodule:: flowvision.models.face_recognition
    :members:
        iresnet50,
        iresnet101,

Semantic Segmentation

.. currentmodule:: flowvision.models

FCN

.. automodule:: flowvision.models.segmentation
    :members:
        fcn_resnet50_coco,
        fcn_resnet101_coco,

DeepLabV3

.. automodule:: flowvision.models.segmentation
    :members:
        deeplabv3_resnet50_coco,
        deeplabv3_resnet101_coco,
        deeplabv3_mobilenet_v3_large_coco,

LRASPP

.. automodule:: flowvision.models.segmentation
    :members:
        lraspp_mobilenet_v3_large_coco,

Object Detection

.. currentmodule:: flowvision.models

Faster R-CNN

.. automodule:: flowvision.models.detection
    :members:
        fasterrcnn_resnet50_fpn,
        fasterrcnn_mobilenet_v3_large_320_fpn,
        fasterrcnn_mobilenet_v3_large_fpn,

RetinaNet

.. automodule:: flowvision.models.detection
    :members:
        retinanet_resnet50_fpn,

SSD

.. automodule:: flowvision.models.detection
    :members:
        ssd300_vgg16,

SSDLite

.. automodule:: flowvision.models.detection
    :members:
        ssdlite320_mobilenet_v3_large,