From 5c9040d6445785ac8fad9d7e0a93266f9a1fec56 Mon Sep 17 00:00:00 2001 From: Ethan Weber Date: Tue, 24 Jan 2023 18:15:51 +0000 Subject: [PATCH] lpips --- nerfstudio/models/instant_ngp.py | 2 +- nerfstudio/models/nerfacto.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/nerfstudio/models/instant_ngp.py b/nerfstudio/models/instant_ngp.py index 80c609c14..ed71920e8 100644 --- a/nerfstudio/models/instant_ngp.py +++ b/nerfstudio/models/instant_ngp.py @@ -137,7 +137,7 @@ class NGPModel(Model): # metrics self.psnr = PeakSignalNoiseRatio(data_range=1.0) self.ssim = structural_similarity_index_measure - self.lpips = LearnedPerceptualImagePatchSimilarity(net_type="vgg") + self.lpips = LearnedPerceptualImagePatchSimilarity(net_type="vgg", normalize=True) def get_training_callbacks( self, training_callback_attributes: TrainingCallbackAttributes diff --git a/nerfstudio/models/nerfacto.py b/nerfstudio/models/nerfacto.py index 42571dda1..d8bf8be9c 100644 --- a/nerfstudio/models/nerfacto.py +++ b/nerfstudio/models/nerfacto.py @@ -219,7 +219,7 @@ class NerfactoModel(Model): # metrics self.psnr = PeakSignalNoiseRatio(data_range=1.0) self.ssim = structural_similarity_index_measure - self.lpips = LearnedPerceptualImagePatchSimilarity(net_type='vgg') + self.lpips = LearnedPerceptualImagePatchSimilarity(net_type='vgg', normalize=True) def get_param_groups(self) -> Dict[str, List[Parameter]]: param_groups = {}