From 31345faed57caa211d63c961087556b46108e17f Mon Sep 17 00:00:00 2001 From: evonneng Date: Thu, 7 Jul 2022 19:06:37 +0000 Subject: [PATCH] clipping images before running metrics --- pyrad/graphs/mipnerf.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pyrad/graphs/mipnerf.py b/pyrad/graphs/mipnerf.py index 6551a53b5..7c8b4459a 100644 --- a/pyrad/graphs/mipnerf.py +++ b/pyrad/graphs/mipnerf.py @@ -169,11 +169,13 @@ class MipNerfGraph(Graph): image = torch.moveaxis(image, -1, 0)[None, ...] rgb_coarse = torch.moveaxis(rgb_coarse, -1, 0)[None, ...] rgb_fine = torch.moveaxis(rgb_fine, -1, 0)[None, ...] + rgb_coarse = torch.clip(rgb_coarse, min=-1, max=1) + rgb_fine = torch.clip(rgb_fine, min=-1, max=1) coarse_psnr = self.psnr(image, rgb_coarse) fine_psnr = self.psnr(image, rgb_fine) fine_ssim = self.ssim(image, rgb_fine) - fine_lpips = self.lpips(image, torch.clip(rgb_fine, min=-1, max=1)) + fine_lpips = self.lpips(image, rgb_fine) writer.put_scalar(name=f"psnr/val_{image_idx}-coarse", scalar=float(coarse_psnr), step=step) writer.put_scalar(name=f"psnr/val_{image_idx}-fine", scalar=float(fine_psnr), step=step)