Skip to main content

Supported Model Architectures

Classification​

  • resnet18
  • resnet26
  • resnet34
  • resnet50
  • resnet101
  • efficientnet_b0
  • efficientnet_b1
  • efficientnet_b2
  • efficientnet_b3
  • efficientnet_b4
  • efficientnet_b5
  • swinv2_base_window8_256
  • swinv2_base_window12_192
  • swinv2_base_window12to16_192to256
  • swinv2_base_window12to24_192to384
  • swinv2_base_window16_256
  • swinv2_cr_small_224
  • swinv2_cr_small_ns_224
  • swinv2_cr_tiny_ns_224
  • swinv2_large_window12_192
  • swinv2_large_window12to16_192to256
  • swinv2_large_window12to24_192to384
  • swinv2_small_window8_256
  • swinv2_small_window16_256
  • swinv2_tiny_window8_256
  • swinv2_tiny_window16_256

Instance Segmentation​

  • maskrcnn_resnet50_fpn
  • maskrcnn_resnet50_fpn_v2

Object Detection​

  • fasterrcnn_resnet50_fpn
  • fasterrcnn_resnet50_fpn_v2
  • fasterrcnn_mobilenet_v3_large_fpn
  • fasterrcnn_mobilenet_v3_large_320_fpn
  • fcos_resnet50_fpn
  • retinanet_resnet50_fpn
  • retinanet_resnet50_fpn_v2
  • ssd300_vgg16
  • ssdlite320_mobilenet_v3_large

Semantic Segmentation​

  • deeplabv3_mobilenet_v3_large
  • deeplabv3_resnet50
  • deeplabv3_resnet101
  • fcn_resnet50
  • fcn_resnet101
  • lraspp_mobilenet_v3_large

Key Point​

  • keypointrcnn_resnet50_fpn