From 068bbbbf24bd04f25949f2da5b44bbe9d5ebe137 Mon Sep 17 00:00:00 2001 From: Ethan Weber Date: Fri, 7 Apr 2023 23:48:20 +0000 Subject: [PATCH] file for masked metrics --- nerfstudio/utils/metrics.py | 8 ++++++++ 1 file changed, 8 insertions(+) create mode 100644 nerfstudio/utils/metrics.py 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