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

annealing

Browse Source
This commit is contained in:
Justin committed 2023-01-30 18:03:00 +00:00
1 parent efd745af13
commit 8ea6534b7e
1 file changed
+3 -4
Download Patch File Download Diff File Expand all files Collapse all files
nerfstudio/models/nerfacto.py
+3 -4
View File
@@ -316,10 +316,9 @@ class NerfactoModel(Model):
outputs["rendered_pred_normal_loss"]
)
#depth regularity loss
if self.step>100:
p=3
patches = outputs['depth'].view(-1,p,p) #Bx3x3
loss_dict['depth_reg'] = .1*torch.mean(torch.sum((patches.view(-1,p*p) - patches[:,p//2,p//2][:,None])**2,dim=1))
p=3
patches = outputs['depth'].view(-1,p,p) #Bx3x3
loss_dict['depth_reg'] = np.interp(self.step,[0,500],[0,1])*torch.mean(torch.sum((patches.view(-1,p*p) - patches[:,p//2,p//2][:,None])**2,dim=1))
return loss_dict