Skip to content

GitLab

Unverified Commit 9aa07224 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Re-enable *RCNN Onnx tests (#3319)

parent c5257708
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
test/test_onnx.py
View file @ 9aa07224
...@@ -368,7 +368,6 @@ class ONNXExporterTester(unittest.TestCase): ...@@ -368,7 +368,6 @@ class ONNXExporterTester(unittest.TestCase):
test_images = [image2] test_images = [image2]
return images, test_images return images, test_images
@unittest.skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged.
def test_faster_rcnn(self): def test_faster_rcnn(self):
images, test_images = self.get_test_images() images, test_images = self.get_test_images()
dummy_image = [torch.ones(3, 100, 100) * 0.3] dummy_image = [torch.ones(3, 100, 100) * 0.3]
...@@ -420,7 +419,6 @@ class ONNXExporterTester(unittest.TestCase): ...@@ -420,7 +419,6 @@ class ONNXExporterTester(unittest.TestCase):
assert torch.all(out2.eq(out_trace2)) 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): def test_mask_rcnn(self):
images, test_images = self.get_test_images() images, test_images = self.get_test_images()
dummy_image = [torch.ones(3, 100, 100) * 0.3] dummy_image = [torch.ones(3, 100, 100) * 0.3]
...@@ -471,7 +469,6 @@ class ONNXExporterTester(unittest.TestCase): ...@@ -471,7 +469,6 @@ class ONNXExporterTester(unittest.TestCase):
assert torch.all(out2[0].eq(out_trace2[0])) assert torch.all(out2[0].eq(out_trace2[0]))
assert torch.all(out2[1].eq(out_trace2[1])) 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): def test_keypoint_rcnn(self):
images, test_images = self.get_test_images() images, test_images = self.get_test_images()
dummy_images = [torch.ones(3, 100, 100) * 0.3] dummy_images = [torch.ones(3, 100, 100) * 0.3]
......
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