Skip to content

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

Notifications You must be signed in to change notification settings

astaka-pe/Dual-DMP

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

92 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Dual Deep Mesh Prior [ECCV2022]

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

  • ubuntu 20.04
  • CUDA 10.2 CUDA 9.0
  • NVIDIA GeForce TITAN X 12GB

1. Installation

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

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}
}