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

file for masked metrics

Browse Source
This commit is contained in:
Ethan Weber 2023-04-07 23:48:20 +00:00
parent c3e67ffd5f
commit 068bbbbf24
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
nerfstudio/utils/metrics.py Normal file
View File

@ -0,0 +1,8 @@
"""
This file contains NeRF metrics with masking capabilities.
"""
from torchmetrics.functional import structural_similarity_index_measure
from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity
# TODO: masked PSNR, SSIM, LPIPS