[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 <[email protected]>
apache · Apr 27, 2021 · 4a4a27b · 4a4a27b
1 parent 3d38067
commit 4a4a27b
Showing 1 changed file with 8 additions and 6 deletions.
diff --git 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):