(self, img, img_gt, sinos)
| 18 | self.dset=dset |
| 19 | self.fp_senet_gt, _, self.ril_odl, __ = self.dset.ril() |
| 20 | def forward(self, img, img_gt, sinos): |
| 21 | sinos_gt=self.radon_senet_gt(img_gt) |
| 22 | sinos_enhanced=self.Sinogram_recon_module(sinos) |
| 23 | img_ril=self.ril(sinos_enhanced) |
| 24 | img_input=torch.cat((img, img_ril), 1) |
| 25 | reconstructed_img=self.Image_recon_module(img_input) |
| 26 | return sinos_gt, sinos_enhanced, img_ril, reconstructed_img |
| 27 | |
| 28 | def ril(self, img): |
| 29 | # return ril results of enhanced sinograms |
nothing calls this directly
no test coverage detected