Skip to content

Latest commit

 

History

History
55 lines (43 loc) · 1.74 KB

README.md

File metadata and controls

55 lines (43 loc) · 1.74 KB
Raw

SCFusion

teaser Authors: Shun-Cheng Wu, Keisuke Tateno, Nassir Navab and Federico Tombari

This is the back-end network of SCFusion repo [paper].

We follow the data structure as the edge-connect repository

Install

# cuda version: nvcc --version
conda create --name scfusion pytorch torchvision tensorboard pillow==6.1 cudatoolkit=YOUR_CUDA_TOOLKIT_VERSION  pyyaml -c pytorch -c conda-forge
conda activate scfusion

Build Meshing tool

The meshing tool allows converting the input/output files into meshes

example

cd extension
python build.py install
  • Train
python train.py --config /pth/to/config.yml
  • test
python test.py --config /pth/to/config.yml
  • sample
    need to install meshing tool extension
python sample.py --config /pth/to/config.yml
  • trace
    export the trained model to be used on c++
python trace.py --config /pth/to/config.yml

The pre-trained model can be downloaded here.

Acknowledgement

This work is supported by the German Research Foundation (DFG, project number 407378162) and the Bavarian State Ministry of Education, Science and the Arts in the framework of the Centre Digitisation Bavaria (ZD.B).