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
# 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
The meshing tool allows converting the input/output files into meshes
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.
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).