Skip to content

Python implementation of SO2, SE2, SO3, and SE3 matrix Lie groups using numpy or pytorch

License

Notifications You must be signed in to change notification settings

utiasSTARS/liegroups

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

92 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

liegroups

Python implementation of SO2, SE2, SO3, and SE3 matrix Lie groups using numpy or PyTorch.

[Documentation]

Installation

To install, cd into the repository directory (the one with setup.py) and run:

pip install .

or

pip install -e .

The -e flag tells pip to install the package in-place, which lets you make changes to the code without having to reinstall every time. Do not do this on shared workstations!

Testing

Ensure you have pytest installed on your system, or install it using conda install pytest or pip install pytest. Then run pytest in the repository directory.

Usage

Numpy and torch implementations are accessible through the liegroups.numpy and liegroups.torch subpackages. By default, the numpy implementation is available through the top-level package.

Access the numpy implementation using something like

from liegroups import SE3

or

from liegroups.numpy import SO2

Access the pytorch implementation using something like

from liegroups.torch import SE2

About

Python implementation of SO2, SE2, SO3, and SE3 matrix Lie groups using numpy or pytorch

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages