@@ -368,7 +368,6 @@ def get_test_images(self):
368
368
test_images = [image2 ]
369
369
return images , test_images
370
370
371
- @unittest .skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged.
372
371
def test_faster_rcnn (self ):
373
372
images , test_images = self .get_test_images ()
374
373
dummy_image = [torch .ones (3 , 100 , 100 ) * 0.3 ]
@@ -420,7 +419,6 @@ def test_paste_mask_in_image(self):
420
419
421
420
assert torch .all (out2 .eq (out_trace2 ))
422
421
423
- @unittest .skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged.
424
422
def test_mask_rcnn (self ):
425
423
images , test_images = self .get_test_images ()
426
424
dummy_image = [torch .ones (3 , 100 , 100 ) * 0.3 ]
@@ -471,7 +469,6 @@ def test_heatmaps_to_keypoints(self):
471
469
assert torch .all (out2 [0 ].eq (out_trace2 [0 ]))
472
470
assert torch .all (out2 [1 ].eq (out_trace2 [1 ]))
473
471
474
- @unittest .skip # Skip until pytorch/vision#3251 is fixed and pytorch/pytorch#50910 is merged.
475
472
def test_keypoint_rcnn (self ):
476
473
images , test_images = self .get_test_images ()
477
474
dummy_images = [torch .ones (3 , 100 , 100 ) * 0.3 ]
0 commit comments