Skip to content

Commit 9aa0722

Browse files
authored
Re-enable *RCNN Onnx tests (#3319)
1 parent c525770 commit 9aa0722

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

test/test_onnx.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -368,7 +368,6 @@ def get_test_images(self):
368368
test_images = [image2]
369369
return images, test_images
370370

371-
@unittest.skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged.
372371
def test_faster_rcnn(self):
373372
images, test_images = self.get_test_images()
374373
dummy_image = [torch.ones(3, 100, 100) * 0.3]
@@ -420,7 +419,6 @@ def test_paste_mask_in_image(self):
420419

421420
assert torch.all(out2.eq(out_trace2))
422421

423-
@unittest.skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged.
424422
def test_mask_rcnn(self):
425423
images, test_images = self.get_test_images()
426424
dummy_image = [torch.ones(3, 100, 100) * 0.3]
@@ -471,7 +469,6 @@ def test_heatmaps_to_keypoints(self):
471469
assert torch.all(out2[0].eq(out_trace2[0]))
472470
assert torch.all(out2[1].eq(out_trace2[1]))
473471

474-
@unittest.skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged.
475472
def test_keypoint_rcnn(self):
476473
images, test_images = self.get_test_images()
477474
dummy_images = [torch.ones(3, 100, 100) * 0.3]

0 commit comments

Comments
 (0)