sifeng666 / optimizemvs Goto Github PK

Created by Yi Wei, Shaohui Liu and Wang Zhao from Tsinghua University.

This repository contains source code for Conditional Single-view Shape Generation for Multi-view Stereo Reconstruction in tensorflow.

The code has been tested with Python 2.7, tensorflow 1.3.0 on Ubuntu 16.04.

git clone https://github.com/weiyithu/OptimizeMVS.git

Python virtual environment is recommended.

cd OptimizeMVS
virtualenv env
source ./env/bin/activate
pip install -r requirements.txt

You need to compile the external libraries:

sh compile.sh

sh init_data.sh

sh train.sh

Default training options are stored in the config folder.

You can download our pretrained model: single-category, multi-category and move it to a folder named demo for evaluation.

sh download_trained_model.sh

To evaluate your own model, set the 'load' option in test.sh as the path to your model.

sh test.sh

The results might be slightly better than reported in the paper.

Part of the external operators are borrowed from latent_3d_points and PointNet++. The multi-view images were rendered from ShapeNetCore with the preprocessing scripts in mvcSnP and the point cloud data was from latent_3d_points. We sincerely thank the authors for their kind help.

This work was supported in part by the National Natural Science Foundation of China under Grant U1813218, Grant 61822603, Grant U1713214, Grant 61672306, and Grant 61572271.

If you find this work useful in your research, please consider citing:

@inproceedings{wei2019conditional,
  author = {Wei, Yi and Liu, Shaohui and Zhao, Wang and Lu, Jiwen and Zhou, Jie},
  title = {Conditional Single-view Shape Generation for Multi-view Stereo Reconstruction},
  booktitle = {Proceedings of IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
  year = {2019}
}

The first three authors share equal contributions.