nodes/MrForExample-ComfyUI-3D-Pack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #72 from jclarkk/main

Browse Source 
Perform laplacian smoothing when fitting mesh
This commit is contained in:
Mr. For Example
2024-03-03 20:37:24 +01:00
committed by GitHub
parent 85f4429910 2bece9cb81
commit 408fca9258
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
algorithms/nerf_marching_cubes_converter.py
+2 -2
View File
@@ -288,10 +288,10 @@ class GSConverterNeRFMarchingCubes(nn.Module):
image_pred, alpha_pred = self.render_mesh(pose)
loss_mse = F.mse_loss(image_pred, image_gt) + 0.1 * F.mse_loss(alpha_pred, alpha_gt)
# loss_lap = laplacian_smooth_loss(self.v + self.deform, self.f)
loss_lap = laplacian_smooth_loss(self.v + self.deform, self.f)
loss_normal = normal_consistency(self.v + self.deform, self.f)
loss_offsets = (self.deform ** 2).sum(-1).mean()
loss = loss_mse + 0.001 * loss_normal + 0.1 * loss_offsets
loss = loss_mse + 0.001 * loss_normal + 0.1 * loss_offsets * loss_lap * 0.01
loss.backward()