Skip to content
[CVPR 2019] Single-Image Piece-wise Planar 3D Reconstruction via Associative Embedding
Branch: master
Clone or download
Latest commit bf534e4 Mar 10, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
configs add inference Mar 6, 2019
data_tools init Mar 4, 2019
misc init Mar 4, 2019
models init Mar 4, 2019
utils init Mar 4, 2019
.gitignore add config Mar 4, 2019
LICENSE Initial commit Feb 25, 2019
README.md Fix a typo Mar 10, 2019
bin_mean_shift.py init Mar 4, 2019
instance_parameter_loss.py init Mar 4, 2019
main.py init Mar 4, 2019
match_segmentation.py init Mar 4, 2019
modules.py init Mar 4, 2019
predict.py add inference Mar 6, 2019
requirements.txt Update requirements.txt Mar 4, 2019

README.md

PlanarReconstruction

PyTorch implementation of our CVPR 2019 paper:

Single-Image Piece-wise Planar 3D Reconstruction via Associative Embedding

Zehao Yu*, Jia Zheng*, Dongze Lian, Zihan Zhou, Shenghua Gao

(* Equal Contribution)

Prepare data

Please download the .tfrecords files for training and testing converted by Chen Liu from here. Then convert the .tfrecords to .npz files as the following:

python data_tools/convert_tfrecords.py --data_type=train --input_tfrecords_file=*train.tfrecords --output_dir=/path/to/save/processd/data
python data_tools/convert_tfrecords.py --data_type=val --input_tfrecords_file=*val.tfrecords --output_dir=/path/to/save/processd/data

Train

Run the following command to train our network.

python main.py train with dataset.root_dir=/path/to/save/processd/data

Evaluation

Please download our trained network from here.

To evaluate the performance of our method, please run:

python main.py eval with dataset.root_dir=/path/to/save/processd/data resume_dir=pretrained.pt dataset.batch_size=1

Prediction

Run the following command to predict on a single image.

python predict.py eval with resume_dir=pretrained.pt input_image=/path/to/image 

Acknowledgements

We thank Chen Liu for his great works and repos.

Citation

Please cite our paper for any purpose of usage.

@inproceedings{Yu2019Single,
  title={Single-Image Piece-wise Planar 3D Reconstruction via Associative Embedding},
  author={Zehao Yu and Jia Zheng and Dongze Lian and Zihan Zhou and Shenghua Gao},
  booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
  year={2019}
}
You can’t perform that action at this time.