From d30c70af5d86b507078f0b2c386f269322df015b Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Wed, 27 Jan 2021 15:28:34 +0000 Subject: [PATCH] Temporarily disable broken RCNN Onnx tests. --- test/test_onnx.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/test/test_onnx.py b/test/test_onnx.py index b2a7624fc61..a57fca1e3d2 100644 --- a/test/test_onnx.py +++ b/test/test_onnx.py @@ -368,6 +368,7 @@ def get_test_images(self): test_images = [image2] return images, test_images + @unittest.skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged. def test_faster_rcnn(self): images, test_images = self.get_test_images() dummy_image = [torch.ones(3, 100, 100) * 0.3] @@ -419,6 +420,7 @@ def test_paste_mask_in_image(self): assert torch.all(out2.eq(out_trace2)) + @unittest.skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged. def test_mask_rcnn(self): images, test_images = self.get_test_images() dummy_image = [torch.ones(3, 100, 100) * 0.3] @@ -469,6 +471,7 @@ def test_heatmaps_to_keypoints(self): assert torch.all(out2[0].eq(out_trace2[0])) assert torch.all(out2[1].eq(out_trace2[1])) + @unittest.skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged. def test_keypoint_rcnn(self): images, test_images = self.get_test_images() dummy_images = [torch.ones(3, 100, 100) * 0.3]