.. role:: hidden :class: hidden-section gluoncv.model_zoo ================= .. automodule:: gluoncv.model_zoo .. currentmodule:: gluoncv.model_zoo gluoncv.model_zoo.get_model ^^^^^^^^^^^^^^^^^^^^^^^^^^^ Returns a pre-defined GluonCV model by name. .. hint:: This is the recommended method for getting a pre-defined model. It support directly loading models from `Gluon Model Zoo `_ as well. .. autosummary:: :nosignatures: get_model Image Classification ^^^^^^^^^^^^^^^^^^^^ :hidden:`CIFAR` ~~~~~~~~~~~~~~~ .. autosummary:: :nosignatures: get_cifar_resnet cifar_resnet20_v1 cifar_resnet56_v1 cifar_resnet110_v1 cifar_resnet20_v2 cifar_resnet56_v2 cifar_resnet110_v2 get_cifar_wide_resnet cifar_wideresnet16_10 cifar_wideresnet28_10 cifar_wideresnet40_8 :hidden:`ImageNet` ~~~~~~~~~~~~~~~~~~ We apply dilattion strategy to pre-trained ResNet models (with stride of 8). Please see :class:`gluoncv.model_zoo.SegBaseModel` for how to use it. .. currentmodule:: gluoncv.model_zoo .. autosummary:: :nosignatures: ResNetV1b resnet18_v1b resnet34_v1b resnet50_v1b resnet101_v1b resnet152_v1b :hidden:`ResNext` ~~~~~~~~~~~~~~~~~~ .. currentmodule:: gluoncv.model_zoo .. autosummary:: :nosignatures: ResNext get_resnext resnext50_32x4d resnext101_32x4d resnext101_64x4d resnext101b_64x4d se_resnext50_32x4d se_resnext101_32x4d se_resnext101_64x4d se_resnext101e_64x4d :hidden:`ResNeSt` ~~~~~~~~~~~~~~~~~~ .. currentmodule:: gluoncv.model_zoo .. autosummary:: :nosignatures: ResNeSt resnest14 resnest26 resnest50 resnest101 resnest200 resnest269 :hidden:`MobileNet` ~~~~~~~~~~~~~~~~~~ .. currentmodule:: gluoncv.model_zoo .. autosummary:: :nosignatures: MobileNet MobileNetV2 get_mobilenet get_mobilenet_v2 mobilenet1_0 mobilenet0_75 mobilenet0_5 mobilenet0_25 mobilenetv2_1_0 mobilenetv2_0_75 mobilenetv2_0_5 mobilenetv2_0_25 mobilenetv3_large mobilenetv3_small :hidden:`SqueezeNet` ~~~~~~~~~~~~~~~~~~ .. currentmodule:: gluoncv.model_zoo .. autosummary:: :nosignatures: SqueezeNet squeezenet1_0 squeezenet1_1 :hidden:`DenseNet` ~~~~~~~~~~~~~~~~~~ .. currentmodule:: gluoncv.model_zoo .. autosummary:: :nosignatures: DenseNet densenet121 densenet161 densenet169 densenet201 Object Detection ^^^^^^^^^^^^^^^^ :hidden:`SSD` ~~~~~~~~~~~~~ .. currentmodule:: gluoncv.model_zoo .. autosummary:: :nosignatures: SSD get_ssd ssd_300_vgg16_atrous_voc ssd_300_vgg16_atrous_coco ssd_300_vgg16_atrous_custom ssd_512_vgg16_atrous_voc ssd_512_vgg16_atrous_coco ssd_512_vgg16_atrous_custom ssd_512_resnet50_v1_voc ssd_512_resnet50_v1_coco ssd_512_resnet50_v1_custom ssd_512_resnet101_v2_voc ssd_512_resnet152_v2_voc VGGAtrousExtractor get_vgg_atrous_extractor vgg16_atrous_300 vgg16_atrous_512 :hidden:`Faster RCNN` ~~~~~~~~~~~~~~~~~~~~~ .. autosummary:: :nosignatures: FasterRCNN get_faster_rcnn faster_rcnn_resnet50_v1b_voc faster_rcnn_resnet50_v1b_coco faster_rcnn_resnet50_v1b_custom :hidden:`YOLOv3` ~~~~~~~~~~~~~~~~ .. autosummary:: :nosignatures: YOLOV3 get_yolov3 yolo3_darknet53_voc yolo3_darknet53_coco yolo3_darknet53_custom Instance Segmentation ^^^^^^^^^^^^^^^^^^^^^ .. currentmodule:: gluoncv.model_zoo :hidden:`Mask RCNN` ~~~~~~~~~~~~~~~~~~~ .. autosummary:: :nosignatures: MaskRCNN get_mask_rcnn mask_rcnn_resnet50_v1b_coco Semantic Segmentation ^^^^^^^^^^^^^^^^^^^^^ .. currentmodule:: gluoncv.model_zoo :hidden:`FCN` ~~~~~~~~~~~~~ .. autosummary:: :nosignatures: FCN get_fcn get_fcn_resnet50_voc get_fcn_resnet101_voc get_fcn_resnet101_coco get_fcn_resnet50_ade get_fcn_resnet101_ade :hidden:`PSPNet` ~~~~~~~~~~~~~~~~ .. autosummary:: :nosignatures: PSPNet get_psp get_psp_resnet101_coco get_psp_resnet101_voc get_psp_resnet50_ade get_psp_resnet101_ade :hidden:`DeepLabV3` ~~~~~~~~~~~~~~~~~~~ .. autosummary:: :nosignatures: DeepLabV3 get_deeplab get_deeplab_resnet101_coco get_deeplab_resnet101_voc get_deeplab_resnet50_ade get_deeplab_resnet101_ade Action Recognition ^^^^^^^^^^^^^^^^^^ .. currentmodule:: gluoncv.model_zoo :hidden:`TSN` ~~~~~~~~~~~~~ .. autosummary:: :nosignatures: vgg16_ucf101 vgg16_hmdb51 vgg16_kinetics400 vgg16_sthsthv2 inceptionv1_ucf101 inceptionv1_hmdb51 inceptionv1_kinetics400 inceptionv1_sthsthv2 inceptionv3_ucf101 inceptionv3_hmdb51 inceptionv3_kinetics400 inceptionv3_sthsthv2 resnet18_v1b_sthsthv2 resnet34_v1b_sthsthv2 resnet50_v1b_sthsthv2 resnet101_v1b_sthsthv2 resnet152_v1b_sthsthv2 resnet18_v1b_kinetics400 resnet34_v1b_kinetics400 resnet50_v1b_kinetics400 resnet101_v1b_kinetics400 resnet152_v1b_kinetics400 resnet50_v1b_ucf101 resnet50_v1b_hmdb51 resnet50_v1b_custom :hidden:`C3D` ~~~~~~~~~~~~~~~~ .. autosummary:: :nosignatures: C3D c3d_kinetics400 :hidden:`I3D` ~~~~~~~~~~~~~~~~~~~ .. autosummary:: :nosignatures: I3D_InceptionV1 i3d_inceptionv1_kinetics400 I3D_InceptionV3 i3d_inceptionv3_kinetics400 I3D_ResNetV1 i3d_resnet50_v1_kinetics400 i3d_resnet101_v1_kinetics400 i3d_nl5_resnet50_v1_kinetics400 i3d_nl10_resnet50_v1_kinetics400 i3d_nl5_resnet101_v1_kinetics400 i3d_nl10_resnet101_v1_kinetics400 i3d_resnet50_v1_sthsthv2 i3d_resnet50_v1_hmdb51 i3d_resnet50_v1_ucf101 i3d_resnet50_v1_custom :hidden:`P3D` ~~~~~~~~~~~~~~~~ .. autosummary:: :nosignatures: P3D p3d_resnet50_kinetics400 p3d_resnet101_kinetics400 :hidden:`R2+1D` ~~~~~~~~~~~~~~~~ .. autosummary:: :nosignatures: R2Plus1D r2plus1d_resnet18_kinetics400 r2plus1d_resnet34_kinetics400 r2plus1d_resnet50_kinetics400 r2plus1d_resnet101_kinetics400 r2plus1d_resnet152_kinetics400 :hidden:`SlowFast` ~~~~~~~~~~~~~~~~ .. autosummary:: :nosignatures: SlowFast slowfast_4x16_resnet50_kinetics400 slowfast_8x8_resnet50_kinetics400 slowfast_4x16_resnet101_kinetics400 slowfast_8x8_resnet101_kinetics400 slowfast_16x8_resnet101_kinetics400 slowfast_16x8_resnet101_50_50_kinetics400 slowfast_4x16_resnet50_custom API Reference ^^^^^^^^^^^^^ Network definitions of GluonCV models .. currentmodule:: gluoncv.model_zoo .. automodule:: gluoncv.model_zoo :members: :imported-members: