tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 10:34:29 +08:00
Code

clipping images before running metrics

Browse Source
This commit is contained in:
evonneng 2022-07-07 19:06:37 +00:00
parent d0d2fbe4a0
commit 31345faed5
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pyrad/graphs/mipnerf.py
View File

@ -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)