Commit 5b1716a2 authored by Lara Haidar's avatar Lara Haidar Committed by Francisco Massa
Browse files

enable mask rcnn test (#1613)

parent 227027d5
......@@ -54,7 +54,7 @@ before_install:
- pip install typing
- |
if [[ $TRAVIS_PYTHON_VERSION == 3.6 ]]; then
pip install onnxruntime
pip install -q --user -i https://test.pypi.org/simple/ ort-nightly==1.0.0.dev1123
fi
- conda install av -c conda-forge
......
......@@ -329,11 +329,10 @@ class ONNXExporterTester(unittest.TestCase):
assert torch.all(out2.eq(out_trace2))
@unittest.skip("Disable test until Resize opset 11 is implemented in ONNX Runtime")
def test_mask_rcnn(self):
images, test_images = self.get_test_images()
model = models.detection.mask_rcnn.maskrcnn_resnet50_fpn(pretrained=True)
model = models.detection.mask_rcnn.maskrcnn_resnet50_fpn(pretrained=True, min_size=200, max_size=300)
model.eval()
model(images)
self.run_model(model, [(images,), (test_images,)])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment