Skip to content

Commit 5b1716a

Browse files
Lara Haidarfmassa
Lara Haidar
authored andcommitted
enable mask rcnn test (#1613)
1 parent 227027d commit 5b1716a

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

.travis.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ before_install:
5454
- pip install typing
5555
- |
5656
if [[ $TRAVIS_PYTHON_VERSION == 3.6 ]]; then
57-
pip install onnxruntime
57+
pip install -q --user -i https://test.pypi.org/simple/ ort-nightly==1.0.0.dev1123
5858
fi
5959
- conda install av -c conda-forge
6060

test/test_onnx.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -329,11 +329,10 @@ def test_paste_mask_in_image(self):
329329

330330
assert torch.all(out2.eq(out_trace2))
331331

332-
@unittest.skip("Disable test until Resize opset 11 is implemented in ONNX Runtime")
333332
def test_mask_rcnn(self):
334333
images, test_images = self.get_test_images()
335334

336-
model = models.detection.mask_rcnn.maskrcnn_resnet50_fpn(pretrained=True)
335+
model = models.detection.mask_rcnn.maskrcnn_resnet50_fpn(pretrained=True, min_size=200, max_size=300)
337336
model.eval()
338337
model(images)
339338
self.run_model(model, [(images,), (test_images,)])

0 commit comments

Comments
 (0)