Skip to content

aboulch/ConvPoint

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

32 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ConvPoint: Generalizing discrete convolutions for unstructured point clouds

SnapNet products

Updates

Major performance update: by reformulating the convolutional layer using matrix mulitplications, the memory consumption has been highly reduced.

Major interface update: the spatial relations are now computed in the network class. The framework is then easier to use and more flexible.

Introduction

This repository propose python scripts for point cloud classification and segmentation. The library is coded with PyTorch.

The journal paper is here: http://www.sciencedirect.com/science/article/pii/S0097849320300224

The conference paper is here: https://diglib.eg.org/handle/10.2312/3dor20191064

A preprint of the paper can be found on Arxiv:
http://arxiv.org/abs/1904.02375

License

Code is released under dual license depending on applications, research or commercial. Reseach license is GPLv3. See the license.

Citation

If you use this code in your research, please consider citing:

@article{BOULCH202024,
title = "ConvPoint: Continuous convolutions for point cloud processing",
journal = "Computers & Graphics",
volume = "88",
pages = "24 - 34",
year = "2020",
issn = "0097-8493",
doi = "https://doi.org/10.1016/j.cag.2020.02.005",
url = "http://www.sciencedirect.com/science/article/pii/S0097849320300224",
author = "Alexandre Boulch",
}

Platform

The code was tested on Ubuntu 16.04 with Anaconda.

Dependencies

  • Pytorch
  • Scikit-learn for confusion matrix computation, and efficient neighbors search
  • TQDM for progress bars
  • PlyFile
  • H5py

All these dependencies can be install via conda in an Anaconda environment or via pip.

The library

Nearest neighbor module

The nearest_neighbors directory contains a very small wrapper for NanoFLANN with OpenMP. To compile the module:

cd nearest_neighbors
python setup.py install --home="."

In the case, you do not want to use this C++/Python wrapper. You still can use the previous version of the nearest neighbors computation with Scikit Learn and Multiprocessing, python only version (slower). To do so, add the following lines at the start of your main script (e.g. modelnet_classif.py):

from global_tags import GlobalTags
GlobalTags.legacy_layer_base(True)

Examples