Qt and Pytorch implementation for our paper "GCN-Denoiser: Mesh Denoising with Graph Convolutional Networks" (ACM Transactions on Graphics 2022)
We propose GCN-Denoiser, a novel feature-preserving mesh denoising method based on graph convolutional networks (GCNs). Unlike previous learning-based mesh denoising methods that exploit hand-crafted or voxel-based representations for feature learning, our method explores the structure of a triangular mesh itself and introduces a graph representation followed by graph convolution operations in the dual space of triangles. We also create a new dataset called PrintData containing 20 real scans with their corresponding ground truths for the research community.
More details and updates can be found in this repo.
- Hardware: Personal computer with NVIDIA GPU.
- Environments: CUDA10.0, Windows system (network training part can also be used on Linux).
- Pytroch C++ 1.2.0 , Eigen, Flann and OpenMesh at runtime.
- Pytorch 1.2.0, numpy, Scipy 1.4.1 and tensorbordx 1.13 (>python3.5) in training stage.
The training code and part of validation data are supplied. Network test can be run by:
cd DenoisingGCN/testSamples
unzip bunny_0_2.zip
cd ../
python datautils.py
python test.py
bunny_0_2/*.mat
are sampled patches from the noisy bunny model with 0.2 level of Gaussian noise.
Executable demo, the corresponding code, and some sampled meshes are supplied.
-
For .exe, windows platform is required and the CUDA PATH must be set in the system environment. Some
.dll
are required (CUDA&LibTorch: c10.dll, c10_cuda.dll, caffe2_nvrtc.dll, nvToolsExt61_1.dll, torch.dll; Qt: Qt5Core.dll, Qt5Gui.dll, Qt5OpenGL.dll, Qt5Widgets.dll). -
For code, Visual Studio 2017 and Qt 5.12 are required.
One version of GCN pre-trained model for synthetic models is supplied.
The printed dataset can be downloaded from the author's personal repo.
If you find this useful for your research, please cite the following paper.
@article{shen2022gcndenoiser,
title={GCN-Denoiser: Mesh Denoising with Graph Convolutional Networks},
author={Shen, yuefan and Fu, Hongbo and Du, Zhongshuo and Chen, Xiang and Burnaev, Evgeny and Zorin, Denis and Zhou, Kun and Zheng, Youyi},
journal={ACM Trans. Graph.},
volume={41},
number={1},
issn={0730-0301},
numpages={14},
year={2022}
}
Waiting for updating...