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

Commit

Permalink
added tests, bump gluoncv version
Browse files Browse the repository at this point in the history
  • Loading branch information
Wei Chu committed Apr 23, 2021
1 parent 54aae29 commit 7831786
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 5 deletions.
2 changes: 1 addition & 1 deletion ci/docker/install/ubuntu_onnx.sh
Original file line number Diff line number Diff line change
Expand Up @@ -30,4 +30,4 @@ echo "Installing libprotobuf-dev and protobuf-compiler ..."
apt-get update || true
apt-get install -y libprotobuf-dev protobuf-compiler

pip3 install pytest==6.2.2 pytest-cov==2.11.1 pytest-xdist==2.2.1 protobuf==3.13.0 onnx==1.8.1 Pillow==5.0.0 tabulate==0.7.5 onnxruntime==1.7.0 'numpy>1.16.0,<1.19.0' gluonnlp==0.10.0 gluoncv==0.8.0
pip3 install pytest==6.2.2 pytest-cov==2.11.1 pytest-xdist==2.2.1 protobuf==3.13.0 onnx==1.8.1 Pillow==5.0.0 tabulate==0.7.5 onnxruntime==1.7.0 'numpy>1.16.0,<1.19.0' gluonnlp==0.10.0 gluoncv==0.10.0
9 changes: 5 additions & 4 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 Down Expand Up @@ -389,8 +389,7 @@ def img_segmentation_test_images(tmpdir_factory):
'deeplab_resnet101_ade',
'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 @@ -399,7 +398,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

0 comments on commit 7831786

Please sign in to comment.