Find a file
Laurent FAINSIN d0cdb8e4ee 🎨 black + ruff
2023-05-15 17:18:10 +02:00
cpp_wrappers 🎨 black + ruff 2023-05-15 17:18:10 +02:00
datasetss 🎨 black + ruff 2023-05-15 17:18:10 +02:00
doc ♻️ rename datasets to datasetss since it interferes with hugginface's module 2023-05-15 16:22:48 +02:00
kernels 🎨 black + ruff 2023-05-15 17:18:10 +02:00
models 🎨 black + ruff 2023-05-15 17:18:10 +02:00
utils 🎨 black + ruff 2023-05-15 17:18:10 +02:00
.gitignore 🙈 ignore ./Data folder 2023-05-15 17:14:51 +02:00
INSTALL.md Corrections 2020-04-27 18:46:34 -04:00
LICENSE.txt Add MIT Licence 2023-05-01 20:39:36 +02:00
plot_convergence.py 🎨 black + ruff 2023-05-15 17:18:10 +02:00
README.md 🩹 convert path to relative 2023-05-15 16:24:56 +02:00
test_models.py 🎨 black + ruff 2023-05-15 17:18:10 +02:00
train_ModelNet40.py 🎨 black + ruff 2023-05-15 17:18:10 +02:00
train_NPM3D.py 🎨 black + ruff 2023-05-15 17:18:10 +02:00
train_S3DIS.py 🎨 black + ruff 2023-05-15 17:18:10 +02:00
train_SemanticKitti.py 🎨 black + ruff 2023-05-15 17:18:10 +02:00
visualize_deformations.py 🎨 black + ruff 2023-05-15 17:18:10 +02:00

Intro figure

Created by Hugues THOMAS

Introduction

This repository contains the implementation of Kernel Point Convolution (KPConv) in PyTorch.

KPConv is also available in Tensorflow (original but older implementation).

Another implementation of KPConv is available in PyTorch-Points-3D

KPConv is a point convolution operator presented in our ICCV2019 paper (arXiv). If you find our work useful in your research, please consider citing:

@article{thomas2019KPConv,
    Author = {Thomas, Hugues and Qi, Charles R. and Deschaud, Jean-Emmanuel and Marcotegui, Beatriz and Goulette, Fran{\c{c}}ois and Guibas, Leonidas J.},
    Title = {KPConv: Flexible and Deformable Convolution for Point Clouds},
    Journal = {Proceedings of the IEEE International Conference on Computer Vision},
    Year = {2019}
}

Installation

This implementation has been tested on Ubuntu 18.04 and Windows 10. Details are provided in INSTALL.md.

Experiments

We provide scripts for three experiments: ModelNet40, S3DIS and SemanticKitti. The instructions to run these experiments are in the doc folder.

  • Object Classification: Instructions to train KP-CNN on an object classification task (Modelnet40).

  • Scene Segmentation: Instructions to train KP-FCNN on a scene segmentation task (S3DIS).

  • SLAM Segmentation: Instructions to train KP-FCNN on a slam segmentation task (SemanticKitti).

  • Pretrained models: We provide pretrained weights and instructions to load them.

  • Visualization scripts: For now only one visualization script has been implemented: the kernel deformations display.

Acknowledgment

Our code uses the nanoflann library.

License

Our code is released under MIT License (see LICENSE file for details).

Updates

  • 27/04/2020: Initial release.
  • 27/04/2020: Added NPM3D support thanks to @GeoSur.