PCNN - Point Convolutional Neural Networks by Extension Operators
This repository contains an implementation to the SIGGRAPH 2018 paper: PCNN - Point Convolutional Neural Networks by Extension Operators.
PCNN is a novel framework for applying convolutional neural networks to point clouds. The framework consists of two operators: extension and restriction, mapping point cloud functions to volumetric functions and vise-versa. A point cloud convolution is defined by pull-back of the Euclidean volumetric convolution via an extension-restriction mechanism.
For more details visit: https://arxiv.org/abs/1803.10091.
The code is compatible with python 3.5 + tensorflow 1.8. In addition, the following packages are required:
To run the training procedure on the ModelNet40 classification task:
Training outputs are saved in:
To run evaluation:
The file pointconv.conf containts additional confguration parameters. To train with a different config file:
python train.py --config file_name