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

[v1.x] Add more ONNX model tests #20210

Merged
merged 3 commits into from
Apr 27, 2021
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 8 additions & 6 deletions tests/python-pytest/onnx/test_onnxruntime_cv.py
Original file line number Diff line number Diff line change
Expand Up @@ -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',
Expand All @@ -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):
Expand Down Expand Up @@ -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',
Expand All @@ -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):
Expand Down