(self)
| 58 | return torch.from_numpy(img.transpose(2, 0, 1)) |
| 59 | |
| 60 | def testMaskRCNN(self): |
| 61 | self._test_model("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml") |
| 62 | |
| 63 | @unittest.skipIf(not torch.cuda.is_available(), "CUDA not available") |
| 64 | def testMaskRCNNGPU(self): |
nothing calls this directly
no test coverage detected