liegroups
Python implementation of SO2, SE2, SO3, and SE3 matrix Lie groups using numpy or PyTorch.
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