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