GeoBi-GNN: Geometry-aware Bi-domain Mesh Denoising via Graph Neural Networks

Abstract

Mesh denoising is an essential geometric processing step for raw meshes generated by 3D scanners and depth cameras. It is intended to remove noise while preserving surface intrinsic features of the underlying model. Existing mesh denoising wisdoms of (1) updating mesh vertices directly (referred to as one-step mesh denoising in the spatial domain) or (2) normal filtering followed by vertex updating (referred to as two-step mesh denoising in the normal domain) seldom consider the correlation between vertex updating and normal filtering. This paper proposes a novel end-to-end geometry-aware dual-graph neural network, called GeoBi-GNN, to perform denoising in spatial and normal domains simultaneously. For the first time, we optimize both positions and normals (i.e., dual domains) in a unified framework of GNN, and show the powerful inter-coordination between the dual domains. GeoBi-GNN fully excavates the native dual-graph structure in the mesh, and creates two graph structures for the spatial noise and the normal noise respectively through the adjacency relationship between vertices and surfaces. We design each GNN as a three-layer U-Net architecture to gradually extract multi-scale features from the input graphs. In addition, a specific graph pooling layer with a cascaded weight estimation strategy is designed to improve the robustness and denoising effect. Due to the intuitive relation between the mesh connectivity and the dual graphs, the proposed method is simple to implement. Comprehensive experiments exhibit that the proposed method is significantly superior to the state-of-the-arts of mesh denoising, especially for the large-scale noise and the complex real scans. Our code is available at https://github.com/zhangyk18/GeoBi-GNN.