Skip to content


Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?

Latest commit


Git stats


Failed to load latest commit information.
Latest commit message
Commit time

Pointwise Convolutional Neural Networks

This is the release of the code for the paper `Pointwise Convolutional Neural Networks' in CVPR 2018.


The code is tested in latest Ubuntu 18.04 LTS with CUDA 9.2 and Tensorflow 1.9.

First, we need to compile the convolution operator as follows:

cd tf_ops/conv3p/
chmod 777
./ -a

The result is a dynamic library file named The Python training and evaluation code loads this library for pointwise convolution. By default, the library contains both a CPU and a GPU implementation of the convolution operator. The use_gpu flag in param.json can be set to true to enable the convolution on the GPU.

To train object classification, execute

python [epoch]

To evaluate, execute

python [epoch] 

By default, epoch is 0 if it is not passed as a parameter to the above command. During training, the network is saved after each epoch. You can resume the training if the network was saved before. Just pass the epoch number to the training command.

Similar code structure is adopted for scene segmentation task. For this task, we also provide a re-implementation of PointNet in PyTorch based on the open source implementation by fxia22.

Training Data


If you are using Tensorflow 1.4, you might want to try compiling with instead. It fixes some include paths due to nsync_cv.h, and set the flag _GLIBCXX_USE_CXX11_ABI=0 to make it compatible to libraries compiled with GCC version earlier than 5.1.


As this is a custom convolution operator we built with minimum optimization tricks that we know, you might find it running more slowly than those Tensorflow built-in operators. Despite that, the experiments were done on NVIDIA GTX 1070, GTX 1080, and Titan X (first generation) without big issues.

It will take hours or 1-2 days depending on your setup to finish training for object recognition. For scene segmentation, it might take longer.


This code includes the following third party libraries and data:

  • Scaled exponential linear units (SeLU) for self-normalization in neural network.

  • ModelNet40 data from PointNet

  • Some other utility code from PointNet

  • h5py


Please cite our paper

    title = {Pointwise Convolutional Neural Networks},
    author = {Binh-Son Hua and Minh-Khoi Tran and Sai-Kit Yeung},
    booktitle = {Computer Vision and Pattern Recognition (CVPR)},
    year = {2018}

if you find this useful for your work.

Future work

We made this simple operator with the hope that existing techniques in 2D image understanding tasks can be brought to 3D in a more straightforward manner. More research along this direction is encouraged.

Please contact the authors at if you have any queries.


Code for Pointwise Convolutional Neural Networks, CVPR 2018








No releases published


No packages published