diff --git a/nerfstudio/utils/metrics.py b/nerfstudio/utils/metrics.py new file mode 100644 index 000000000..f9a4b4f66 --- /dev/null +++ b/nerfstudio/utils/metrics.py @@ -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 \ No newline at end of file