(self, use_cuda=True, crop_factor=9 / 16, output_size=(240, 240), padding_factor=0.5)
| 334 | |
| 335 | class SynthTwoPairTnf(SynthPairTnf): |
| 336 | def __init__(self, use_cuda=True, crop_factor=9 / 16, output_size=(240, 240), padding_factor=0.5): |
| 337 | super().__init__(use_cuda=use_cuda) |
| 338 | # self.aff_reorder_idx=torch.LongTensor([3,2,5,1,0,4]) |
| 339 | self.affTnf = GeometricTnf(geometric_model='affine', out_h=self.out_h, out_w=self.out_w, |
| 340 | use_cuda=self.use_cuda) |
| 341 | self.tpsTnf = GeometricTnf(geometric_model='tps', out_h=self.out_h, out_w=self.out_w, |
| 342 | use_cuda=self.use_cuda) |
| 343 | |
| 344 | def __call__(self, batch): |
| 345 | image_batch, theta_batch = batch['image'], batch['theta'] |
nothing calls this directly
no test coverage detected