From 2bece9cb81aa0dcdac75964cd9c1c0124dc91c16 Mon Sep 17 00:00:00 2001 From: Jonathan Clark Date: Sun, 3 Mar 2024 14:10:32 +0200 Subject: [PATCH] Perform laplacian smoothing when fitting mesh --- algorithms/nerf_marching_cubes_converter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/algorithms/nerf_marching_cubes_converter.py b/algorithms/nerf_marching_cubes_converter.py index a402c06..b0513b8 100644 --- a/algorithms/nerf_marching_cubes_converter.py +++ b/algorithms/nerf_marching_cubes_converter.py @@ -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()