From 3dc08247301eab006768aed91d4a8a2ae0647290 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Thu, 28 Jan 2021 15:54:09 +0000 Subject: [PATCH] Revert "Temporarily disable broken RCNN Onnx tests. (#3310)" This reverts commit 691ec6ddd824c4967ae1093fde32133902dd1b70. --- test/test_onnx.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/test/test_onnx.py b/test/test_onnx.py index a57fca1e3d2..b2a7624fc61 100644 --- a/test/test_onnx.py +++ b/test/test_onnx.py @@ -368,7 +368,6 @@ 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] @@ -420,7 +419,6 @@ 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] @@ -471,7 +469,6 @@ 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]