Implementation of CVPR'2022:Surface Reconstruction from Point Clouds by Learning Predictive Context Priors

Last update: Dec 12, 2022

Overview

Surface Reconstruction from Point Clouds by Learning Predictive Context Priors (CVPR 2022)

Personal Web Pages | Paper | Project Page

This repository contains the code to reproduce the results from the paper. [Surface Reconstruction from Point Clouds by Learning Predictive Context Priors](https://arxiv.org/abs/2204.11015).

You can find detailed usage instructions for training your own models and using pretrained models below.

If you find our code or paper useful, please consider citing

@inproceedings{PredictiveContextPriors,
    title = {Surface Reconstruction from Point Clouds by Learning Predictive Context Priors},
    author = {Baorui, Ma and Yu-Shen, Liu and Matthias, Zwicker and Zhizhong, Han},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    year = {2022}
}

Surface Reconstruction Demo

Predicted Queries Visualization

Predicted queries in Loccal Coorinate System

Installation

First you have to make sure that you have all dependencies in place. The simplest way to do so, is to use anaconda.

You can create an anaconda environment called tf using

conda env create -f tf.yaml
conda activate tf

Training

You should train the Local Context Prior Network first, run

python pcp.py --input_ply_file test.ply --data_dir ./data/ --CUDA 0 --OUTPUT_DIR_LOCAL ./local_net/ --OUTPUT_DIR_GLOBAL ./glocal_net/ --train --save_idx -1

You should put the point cloud file(--input_ply_file, only ply format) into the '--data_dir' folder.

Then train the Predictive Context Prior Network, run

python pcp.py --input_ply_file test.ply --data_dir ./data/ --CUDA 0 --OUTPUT_DIR_LOCAL ./local_net/ --OUTPUT_DIR_GLOBAL ./glocal_net/ --finetune --save_idx -1

Test

You can extract the mesh model from the trained network, run

python pcp.py --input_ply_file test.ply --data_dir ./data/ --CUDA 0 --OUTPUT_DIR_LOCAL ./local_net/ --OUTPUT_DIR_GLOBAL ./glocal_net/ --test --save_idx -1

Implementation of CVPR'2022:Surface Reconstruction from Point Clouds by Learning Predictive Context Priors

Related tags

Overview

Surface Reconstruction from Point Clouds by Learning Predictive Context Priors (CVPR 2022)

Personal Web Pages | Paper | Project Page

Surface Reconstruction Demo

Predicted Queries Visualization

Predicted queries in Loccal Coorinate System

Installation

Training

Test

Owner

A PyTorch Implementation of Single Shot MultiBox Detector

Official codebase for Pretrained Transformers as Universal Computation Engines.

Optimal Camera Position for a Practical Application of Gaze Estimation on Edge Devices,

make ASCII Art by Deep Learning

🔥 Cogitare - A Modern, Fast, and Modular Deep Learning and Machine Learning framework for Python

The repository contains reproducible PyTorch source code of our paper Generative Modeling with Optimal Transport Maps, ICLR 2022.

Racing line optimization algorithm in python that uses Particle Swarm Optimization.

Pytorch implementation of Learning with Opponent-Learning Awareness

This is a repository for a Semantic Segmentation inference API using the Gluoncv CV toolkit

This is the code for Deformable Neural Radiance Fields, a.k.a. Nerfies.

This repository contains an overview of important follow-up works based on the original Vision Transformer (ViT) by Google.

ReGAN: Sequence GAN using RE[INFORCE|LAX|BAR] based PG estimators

CS50's Introduction to Artificial Intelligence Test Scripts

Cross-Task Consistency Learning Framework for Multi-Task Learning

Designing a Practical Degradation Model for Deep Blind Image Super-Resolution (ICCV, 2021) (PyTorch) - We released the training code!

Camera-caps - Examine the camera capabilities for V4l2 cameras

Python suite to construct benchmark machine learning datasets from the MIMIC-III clinical database.

Multi-atlas segmentation (MAS) is a promising framework for medical image segmentation

SeqTR: A Simple yet Universal Network for Visual Grounding

A trusty face recognition research platform developed by Tencent Youtu Lab