Method decode_voxel

src/encoding/model.py:476–488

(self, save_dir, triplane_feat, reso, n_faces=10000, only_largest_cc=True)

    @torch.no_grad()
    def decode_voxel(self, save_dir, triplane_feat, reso, n_faces=10000, only_largest_cc=True):
        H, W = triplane_feat[0].shape[-2:]
        D = triplane_feat[1].shape[-1]
        new_aabb = self._resize_aabb((H, W, D))

        # maching cubes on sdf grid
        os.makedirs(save_dir, exist_ok=True)
        sdf_grid = self.decode_grid(triplane_feat, reso, aabb=new_aabb)[..., 0].detach().cpu().numpy() # (H, W, D, 1)
        # sdf_grid = -F.max_pool3d(-sdf_grid[None, None], 2, 2)[0, 0].detach().cpu().numpy()
        vox_grid = sdf_grid < 0

        save_path = os.path.join(save_dir, f"r{reso}_voxel.npz")
        np.savez_compressed(save_path, vox_grid=vox_grid)


def evaluate_tsdf_prediction(pred_sdf, gt_sdf, sdf_threshold):