tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-15 04:13:57 +08:00
Code Activity

lpips

Browse Source
This commit is contained in:
Ethan Weber committed 2023-01-24 18:15:51 +00:00
1 parent ca6c8efbe2
commit 5c9040d644
2 files changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

nerfstudio/models/instant_ngp.py
+1 -1
View File
@@ -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
nerfstudio/models/nerfacto.py
+1 -1
View File
@@ -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 = {}