{"id":33174748,"url":"https://github.com/svip-lab/PlanarReconstruction","last_synced_at":"2025-11-20T20:01:41.949Z","repository":{"id":41259795,"uuid":"172488131","full_name":"svip-lab/PlanarReconstruction","owner":"svip-lab","description":"[CVPR'19] Single-Image Piece-wise Planar 3D Reconstruction via Associative Embedding","archived":false,"fork":false,"pushed_at":"2024-02-10T09:52:46.000Z","size":837,"stargazers_count":369,"open_issues_count":10,"forks_count":88,"subscribers_count":21,"default_branch":"master","last_synced_at":"2025-05-25T19:09:21.745Z","etag":null,"topics":["computer-vision","cvpr2019","deep-learning","pytorch"],"latest_commit_sha":null,"homepage":"","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/svip-lab.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2019-02-25T10:50:31.000Z","updated_at":"2025-05-21T01:11:36.000Z","dependencies_parsed_at":"2024-12-29T20:30:12.307Z","dependency_job_id":null,"html_url":"https://github.com/svip-lab/PlanarReconstruction","commit_stats":null,"previous_names":[],"tags_count":0,"template":false,"template_full_name":null,"purl":"pkg:github/svip-lab/PlanarReconstruction","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/svip-lab%2FPlanarReconstruction","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/svip-lab%2FPlanarReconstruction/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/svip-lab%2FPlanarReconstruction/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/svip-lab%2FPlanarReconstruction/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/svip-lab","download_url":"https://codeload.github.com/svip-lab/PlanarReconstruction/tar.gz/refs/heads/master","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/svip-lab%2FPlanarReconstruction/sbom","scorecard":null,"host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":285503668,"owners_count":27182913,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","status":"online","status_checked_at":"2025-11-20T02:00:05.334Z","response_time":54,"last_error":null,"robots_txt_status":"success","robots_txt_updated_at":"2025-07-24T06:49:26.215Z","robots_txt_url":"https://github.com/robots.txt","online":true,"can_crawl_api":true,"host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["computer-vision","cvpr2019","deep-learning","pytorch"],"created_at":"2025-11-16T02:00:33.687Z","updated_at":"2025-11-20T20:01:41.941Z","avatar_url":"https://github.com/svip-lab.png","language":"Python","funding_links":[],"categories":["References"],"sub_categories":["Papers - Scene level"],"readme":"# PlanarReconstruction\n\nPyTorch implementation of our CVPR 2019 paper:\n\n[Single-Image Piece-wise Planar 3D Reconstruction via Associative Embedding](https://arxiv.org/pdf/1902.09777.pdf)\n\nZehao Yu\\*,\n[Jia Zheng](https://bertjiazheng.github.io/)\\*,\n[Dongze Lian](https://svip-lab.github.io/team/liandz.html),\n[Zihan Zhou](https://faculty.ist.psu.edu/zzhou/Home.html),\n[Shenghua Gao](http://sist.shanghaitech.edu.cn/sist_en/2018/0820/c3846a31775/page.htm)\n\n(\\* Equal Contribution)\n\n\u003cimg src=\"misc/pipeline.jpg\" width=\"800\"\u003e\n\n## Getting Started\n\n### Installation\n\nClone repository and use [git-lfs](https://git-lfs.github.com/) to fetch the trained model (or download [here](https://drive.google.com/file/d/1Aa1Jb0CGpiYXKHeTwpXAwcwu_yEqdkte/view?usp=sharing)):\n```bash\ngit clone git@github.com:svip-lab/PlanarReconstruction.git\n```\n\nWe use Python 3. Create an Anaconda enviroment and install the dependencies:\n```bash\nconda create -y -n plane python=3.6\nconda activate plane\nconda install -c menpo opencv\npip install -r requirements.txt\n```\n\n### Downloading and converting data\nPlease download the *.tfrecords* files for training and testing converted by [PlaneNet](https://github.com/art-programmer/PlaneNet), then convert the *.tfrecords* to *.npz* files:\n```bash\npython data_tools/convert_tfrecords.py --data_type=train --input_tfrecords_file=/path/to/planes_scannet_train.tfrecords --output_dir=/path/to/save/processd/data\npython data_tools/convert_tfrecords.py --data_type=val --input_tfrecords_file=/path/to/planes_scannet_val.tfrecords --output_dir=/path/to/save/processd/data\n```\n\n### Training\nRun the following command to train our network:\n```bash\npython main.py train with dataset.root_dir=/path/to/save/processd/data\n```\n\n### Evaluation\nRun the following command to evaluate the performance:\n```bash\npython main.py eval with dataset.root_dir=/path/to/save/processd/data resume_dir=/path/to/pretrained.pt dataset.batch_size=1\n```\n\n### Prediction\nRun the following command to predict on a single image:\n```bash\npython predict.py with resume_dir=pretrained.pt image_path=/path/to/image\n```\n\n## Acknowledgements\nWe thank [Chen Liu](http://art-programmer.github.io/index.html) for his great works and repos.\n\n## Citation\nPlease cite our paper for any purpose of usage.\n```\n@inproceedings{YuZLZG19,\n  author    = {Zehao Yu and Jia Zheng and Dongze Lian and Zihan Zhou and Shenghua Gao},\n  title     = {Single-Image Piece-wise Planar 3D Reconstruction via Associative Embedding},\n  booktitle = {CVPR},\n  pages     = {1029--1037},\n  year      = {2019}\n}\n```\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fsvip-lab%2FPlanarReconstruction","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fsvip-lab%2FPlanarReconstruction","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fsvip-lab%2FPlanarReconstruction/lists"}