[incubator-mxnet] branch v1.x updated: [v1.x] Add more ONNX model tests (#20210)

[zh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

zha0q1 pushed a commit to branch v1.x
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/v1.x by this push:
     new 4a4a27b  [v1.x] Add more ONNX model tests (#20210)
4a4a27b is described below

commit 4a4a27b25f7c645e5be9a9776fcff53c68e632df
Author: waytrue17 <52...@users.noreply.github.com>
AuthorDate: Tue Apr 27 13:24:21 2021 -0700

    [v1.x] Add more ONNX model tests (#20210)
    
    * added tests, bump gluoncv version
    
    * revert gluoncv version
    
    * fix mask_rcnn test
    
    Co-authored-by: Wei Chu <we...@amazon.com>
---
 tests/python-pytest/onnx/test_onnxruntime_cv.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/tests/python-pytest/onnx/test_onnxruntime_cv.py b/tests/python-pytest/onnx/test_onnxruntime_cv.py
index fd9a323..f0c454e 100644
--- a/tests/python-pytest/onnx/test_onnxruntime_cv.py
+++ b/tests/python-pytest/onnx/test_onnxruntime_cv.py
@@ -259,7 +259,7 @@ def obj_detection_test_images(tmpdir_factory):
     'yolo3_mobilenet1.0_voc',
     'ssd_300_vgg16_atrous_coco',
     'ssd_512_vgg16_atrous_coco',
-    # 'ssd_300_resnet34_v1b_coco', #cannot import
+    'ssd_300_resnet34_v1b_coco',
     'ssd_512_resnet50_v1_coco',
     'ssd_512_mobilenet1.0_coco',
     'faster_rcnn_resnet50_v1b_coco',
@@ -270,11 +270,11 @@ def obj_detection_test_images(tmpdir_factory):
     # Those two models were failing in nightly ci due to anticipated mxnet onnx nms numerical differences.
     # Model ouputs look good to human eyes. We will need to rewrite the bbox check
     #'faster_rcnn_fpn_resnet101_v1d_coco',
-    #'mask_rcnn_resnet18_v1b_coco',
-    'mask_rcnn_fpn_resnet18_v1b_coco',
+    #'mask_rcnn_fpn_resnet18_v1b_coco',
+    'mask_rcnn_resnet18_v1b_coco',
     'mask_rcnn_resnet50_v1b_coco',
-    'mask_rcnn_fpn_resnet50_v1b_coco',
     'mask_rcnn_resnet101_v1d_coco',
+    'mask_rcnn_fpn_resnet50_v1b_coco',
     'mask_rcnn_fpn_resnet101_v1d_coco',
 ])
 def test_obj_detection_model_inference_onnxruntime(tmp_path, model, obj_detection_test_images):
@@ -392,7 +392,7 @@ def img_segmentation_test_images(tmpdir_factory):
     'deeplab_resnest50_ade',
     'deeplab_resnest101_ade',
     # cannot download this model, skipping for now
-    #'deeplab_resnest200_ade',
+    # 'deeplab_resnest200_ade',
     'deeplab_resnest269_ade',
     'fcn_resnet101_coco',
     'deeplab_resnet101_coco',
@@ -401,7 +401,9 @@ def img_segmentation_test_images(tmpdir_factory):
     'deeplab_resnet152_voc',
     pytest.param('deeplab_resnet50_citys', marks=pytest.mark.integration),
     'deeplab_resnet101_citys',
-    'deeplab_v3b_plus_wideresnet_citys'
+    'deeplab_v3b_plus_wideresnet_citys',
+    'danet_resnet50_citys',
+    'danet_resnet101_citys'
 ])
 def test_img_segmentation_model_inference_onnxruntime(tmp_path, model, img_segmentation_test_images):
     def normalize_image(imgfile):