mugdhak21 / dual-dmp

Learning Self-prior for Mesh Denoising using Dual Graph Convolutional Networks [ECCV 2022]

Dual Deep Mesh Prior [ECCV2022]

Paper | Supp | Poster

The official implementation of Learning Self-prior for Mesh Denoising using Dual Graph Convolutional Networks, ECCV2022.

A deep-learning framework for mesh denoising from a single noisy input, where two graph convolutional networks are trained jointly to filter vertex positions and facet normals apart.

Method Overview

Results

---

Getting Started

Tested environment

• 
• 
• 

1. Installation

git clone https://github.com/astaka-pe/Dual-DMP
cd Dual-DMP
conda env create -f environment.yml
conda activate ddmp

for pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html

2. Preparation

The Dataset is distributed as a zip file. Please unzip and place it under Dual-DMP directory.

3. Training

• CAD model

python main.py -i datasets/fandisk --k1 3 --k2 0 --k3 3 --k4 4 --k5 2 --bnfloop 5

• Non-CAD model

python main.py -i datasets/ankylosaurus

• Real-scanned model

python main.py -i datasets/pyramid --iter 50

Outputs will be generated under datasets/{model-name}/output/ with their MAD scores.

---

Appendix

Training with your own data

Place a noisy mesh and a ground-truth mesh under datasets/{model-name}/ .

• Noisy mesh: {model-name}_noise.obj
• Ground-truth mesh: {model-name}_gt.obj

Run

python preprocess/preprocess.py -i datasets/{model-name}

for edge-based normalization and creating initial smoothed mesh.

Finally, run

python main.py -i datasets/{model-name}

You should set appropriate weights as discribed in the paper.

Training without using ground-truth data

After runnning preprocess.py, run

python main4real.py -i datasets/{model-name}

Creating noisy data

Run

python preprocess/noisemaker.py -i datasets/{model-name}/{model-name}.obj --level {noise-level}

---

Citation

@InProceedings{hattori2022ddmp,
  author        = {Hattori, Shota and Yatagawa, Tatsuya and Ohtake, Yutaka and Suzuki, Hiromasa},
  title         = {Learning Self-prior for Mesh Denoising using Dual Graph Convolutional Networks},
  booktitle     = {Proceedings of the European Conference on Computer Vision (ECCV)},
  year          = {2022},
  doi           = {10.1007/978-3-031-20062-5_21}
}