tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-14 03:44:02 +08:00
Code Activity

added psnr to the local writer prints of ngp

Browse Source
This commit is contained in:
evonneng committed 2022-06-11 21:29:16 +00:00
1 parent 201f85a175
commit fb63bb543f
1 file changed
+2
Download Patch File Download Diff File Expand all files Collapse all files
pyrad/graphs/instant_ngp.py
+2
View File
@@ -162,4 +162,6 @@ class NGPGraph(Graph):
writer.put_scalar(name=f"ssim/val_{image_idx}", scalar=float(ssim), step=step)
writer.put_scalar(name=f"lpips/val_{image_idx}", scalar=float(lpips), step=step)
writer.put_scalar(name=writer.EventName.CURR_TEST_PSNR, scalar=float(psnr), step=step)
return psnr.item()