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)