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 = {}