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

Unofficial implement with paper SpeakerGAN: Speaker identification with conditional generative adversarial network

Self-Supervised Speech Pre-training and Representation Learning Toolkit.

Your interactive network visualizing dashboard

OpenMMLab Model Deployment Toolset

Dynamic Attentive Graph Learning for Image Restoration, ICCV2021 [PyTorch Code]

PyTorch implementation of the Value Iteration Networks (VIN) (NIPS '16 best paper)

GNPy: Optical Route Planning and DWDM Network Optimization

Self-Regulated Learning for Egocentric Video Activity Anticipation

PyTorch Lightning + Hydra. A feature-rich template for rapid, scalable and reproducible ML experimentation with best practices. ⚡🔥⚡

Machine learning for NeuroImaging in Python

A python package to perform same transformation to coco-annotation as performed on the image.

RetinaFace: Deep Face Detection Library in TensorFlow for Python

Secure Distributed Training at Scale

PyTorch implementation of SmoothGrad: removing noise by adding noise.

[NeurIPS 2021] "G-PATE: Scalable Differentially Private Data Generator via Private Aggregation of Teacher Discriminators"

Code for the paper A Theoretical Analysis of the Repetition Problem in Text Generation

Program your own vulkan.gpuinfo.org query in Python. Used to determine baseline hardware for WebGPU.

Minimalist Error collection Service compatible with Rollbar clients. Sentry or Rollbar alternative.

Bidimensional Leaderboards: Generate and Evaluate Language Hand in Hand

Stock-history-display - something like a easy yearly review for your stock performance