Skip to main content

Pure Numpy Implementation of the Coherent Point Drift Algorithm

Project description

Pure Numpy Implementation of the Coherent Point Drift Algorithm.

MIT License.

Introduction

This is a pure numpy implementation of the coherent point drift CPD algorithm by Myronenko and Song. It provides three registration methods for point clouds: 1) Scale and rigid registration; 2) Affine registration; and 3) Gaussian regularized non-rigid registration.

The CPD algorithm is a registration method for aligning two point clouds. In this method, the moving point cloud is modelled as a Gaussian Mixture Model (GMM) and the fixed point cloud are treated as observations from the GMM. The optimal transformation parameters maximze the Maximum A Posteriori (MAP) estimation that the observed point cloud is drawn from the GMM.

The registration methods work for 2D and 3D point clouds. For more information, please refer to my blog

Pip Install

$ pip install pycpd

Installation From Source

Clone the repository to a location in your home directory. For example:

$ git clone https://github.com/siavashk/pycpd.git $HOME/pycpd

Install the package:

$ pip install .

For running sample registration examples under /tests, you will need two additional packages.

Scipy (for loading .mat files) and matplotlib (for visualizing the reigstration). These can be downloaded by running:

$ pip install -r requirements.txt

Usage

Each registration method is contained within a single class inside the pycpd subfolder. To try out the registration, you can simply call:

$ python examples/fish{Transform}{Dimension}.py

where Transform is either Rigid, Affine or Deformable and Dimension is either 2D or 3D.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Built Distribution

pycpd-0.3-py2.py3-none-any.whl (9.1 kB view hashes)

Uploaded py2 py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page