Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
[v1.x] Add ONNX model support matrix (#20230)
Browse files Browse the repository at this point in the history
* add model support matrix

* add gluon links

Co-authored-by: Wei Chu <[email protected]>
  • Loading branch information
waytrue17 and Wei Chu authored May 3, 2021
1 parent 25b2f3b commit 987f2d5
Show file tree
Hide file tree
Showing 2 changed files with 394 additions and 2 deletions.
175 changes: 173 additions & 2 deletions python/mxnet/onnx/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -180,6 +180,177 @@ If you are a user of earlier MXNet versions and do not want to upgrade MXNet, yo
|where|1.7 1.8 |
|zeros_like|1.7 1.8 |

### GluonCV Pretrained Model Support Matrix
### [GluonCV Pretrained Model Support Matrix](https://cv.gluon.ai/model_zoo/index.html)
|Image Classification|
|:-|
|alexnet|
|cifar_resnet20_v1|
|cifar_resnet56_v1|
|cifar_resnet110_v1|
|cifar_resnet20_v2|
|cifar_resnet56_v2|
|cifar_resnet110_v2|
|cifar_wideresnet16_10|
|cifar_wideresnet28_10|
|cifar_wideresnet40_8|
|cifar_resnext29_16x64d|
|darknet53|
|densenet121|
|densenet161|
|densenet169|
|densenet201|
|googlenet|
|mobilenet1.0|
|mobilenet0.75|
|mobilenet0.5|
|mobilenet0.25|
|mobilenetv2_1.0|
|mobilenetv2_0.75|
|mobilenetv2_0.5|
|mobilenetv2_0.25|
|mobilenetv3_large|
|mobilenetv3_small|
|resnest14|
|resnest26|
|resnest50|
|resnest101|
|resnest200|
|resnest269|
|resnet18_v1|
|resnet18_v1b_0.89|
|resnet18_v2|
|resnet34_v1|
|resnet34_v2|
|resnet50_v1|
|resnet50_v1d_0.86|
|resnet50_v1d_0.48|
|resnet50_v1d_0.37|
|resnet50_v1d_0.11|
|resnet50_v2|
|resnet101_v1|
|resnet101_v1d_0.76|
|resnet101_v1d_0.73|
|resnet101_v2|
|resnet152_v1|
|resnet152_v2|
|resnext50_32x4d|
|resnext101_32x4d|
|resnext101_64x4d|
|senet_154|
|se_resnext101_32x4d|
|se_resnext101_64x4d|
|se_resnext50_32x4d|
|squeezenet1.0|
|squeezenet1.1|
|vgg11|
|vgg11_bn|
|vgg13|
|vgg13_bn|
|vgg16|
|vgg16_bn|
|vgg19|
|vgg19_bn|
|xception|
|inceptionv3|

### GluonNLP Pretrained Model Support Matrix
|Object Detection|
|:-|
|center_net_resnet18_v1b_voc|
|center_net_resnet50_v1b_voc|
|center_net_resnet101_v1b_voc|
|center_net_resnet18_v1b_coco|
|center_net_resnet50_v1b_coco|
|center_net_resnet101_v1b_coco|
|ssd_300_vgg16_atrous_voc|
|ssd_512_vgg16_atrous_voc|
|ssd_512_resnet50_v1_voc|
|ssd_512_mobilenet1.0_voc|
|faster_rcnn_resnet50_v1b_voc|
|yolo3_darknet53_voc|
|yolo3_mobilenet1.0_voc|
|ssd_300_vgg16_atrous_coco|
|ssd_512_vgg16_atrous_coco|
|ssd_300_resnet34_v1b_coco|
|ssd_512_resnet50_v1_coco|
|ssd_512_mobilenet1.0_coco|
|faster_rcnn_resnet50_v1b_coco|
|faster_rcnn_resnet101_v1d_coco|
|yolo3_darknet53_coco|
|yolo3_mobilenet1.0_coco|
|faster_rcnn_fpn_resnet50_v1b_coco|
|faster_rcnn_fpn_resnet101_v1d_coco|
|mask_rcnn_fpn_resnet18_v1b_coco|
|mask_rcnn_resnet18_v1b_coco|
|mask_rcnn_resnet50_v1b_coco|
|mask_rcnn_resnet101_v1d_coco|
|mask_rcnn_fpn_resnet50_v1b_coco|
|mask_rcnn_fpn_resnet101_v1d_coco|

|Image Segmentation|
|:-|
|fcn_resnet50_ade|
|fcn_resnet101_ade|
|deeplab_resnet50_ade|
|deeplab_resnet101_ade|
|deeplab_resnest50_ade|
|deeplab_resnest101_ade|
|deeplab_resnest269_ade|
|fcn_resnet101_coco|
|deeplab_resnet101_coco|
|fcn_resnet101_voc|
|deeplab_resnet101_voc|
|deeplab_resnet152_voc|
|deeplab_resnet50_citys|
|deeplab_resnet101_citys|
|deeplab_v3b_plus_wideresnet_citys|
|danet_resnet50_citys|
|danet_resnet101_citys|

|Pose Estimation|
|:-|
|simple_pose_resnet18_v1b|
|simple_pose_resnet50_v1b|
|simple_pose_resnet50_v1d|
|simple_pose_resnet101_v1b|
|simple_pose_resnet101_v1d|
|simple_pose_resnet152_v1b|
|simple_pose_resnet152_v1d|
|alpha_pose_resnet101_v1b_coco|
|mobile_pose_resnet18_v1b|
|mobile_pose_resnet50_v1b|
|mobile_pose_mobilenet1.0|
|mobile_pose_mobilenetv2_1.0|
|mobile_pose_mobilenetv3_large|
|mobile_pose_mobilenetv3_small|

|Action Estimation|
|:-|
|inceptionv1_kinetics400|
|resnet18_v1b_kinetics400|
|resnet34_v1b_kinetics400|
|resnet50_v1b_kinetics400|
|resnet101_v1b_kinetics400|
|resnet152_v1b_kinetics400|
|resnet50_v1b_hmdb51|
|resnet50_v1b_sthsthv2|
|vgg16_ucf101|
|inceptionv3_kinetics400|
|inceptionv3_ucf101|

### [GluonNLP Pretrained Model Support Matrix](https://nlp.gluon.ai/model_zoo/catalog.html)
|NLP Models|
|:-|
|awd_lstm_lm_600|
|awd_lstm_lm_1150|
|bert_12_768_12|
|bert_24_1024_16|
|distilbert_6_768_12|
|ernie_12_768_12|
|gpt2_117m|
|gpt2_345m|
|roberta_12_768_12|
|roberta_24_1024_16|
|standard_lstm_lm_200|
|standard_lstm_lm_650|
|standard_lstm_lm_1500|
|transformer_en_de_512|
Loading

0 comments on commit 987f2d5

Please sign in to comment.