(self)
| 52 | self.fid.reset() |
| 53 | |
| 54 | def aggregate(self): |
| 55 | fid = self.fid.compute() |
| 56 | ssim = torch.stack(self.ssims).mean() |
| 57 | psnr = torch.stack(self.psnrs).mean() |
| 58 | out = dict(fid=fid, ssim=ssim, psnr=psnr) |
| 59 | return out |
| 60 | |
| 61 | |
| 62 | class DepthMetricTracker: |
no outgoing calls
no test coverage detected