A TensorFlow-powered python toolbox to train deep neural networks to perform motor tasks.
Project description
MotorNet
This repository contains MotorNet, a python package that allows training recurrent neural networks to control for biomechanically realistic effectors. This toolbox is designed to meet several goals:
- No hard dependencies beyond typically available python packages. This should make it easier to run this package on remote computing units.
- Provide users with a variety of muscle types to chose from.
- Flexibility in creating new, arbitrary muscle wrappings around the skeleton, to enable fast exploration of different potential effectors and how to control them. The moment arm are calculated online according to the geometry of the skeleton and the (user-defined) paths of the muscles.
- User-friendly API, to allow for easier familiarization and intuitive usage. We want to enable a focus on ideas, not implementation. The toolbox focuses on subclassing to allow users to implement their custom task designs, custom plants, and custom controller networks.
- Open-source, to allow for user contribution and community-driven incremental progress.
State of the project
The package is feature complete, and is used by several people to progress in their research. We are currently in beta phase, meaning the toolbox is publicly available but we are still on the lookout for potential bugs and fixes to apply. Please feel free to log an issue if you think you found a bug, we appreciate any contribution. Stay tuned for more!
An online documentation is available. Feel free to check it out.
How to Install
Install with pip
First, please make sure that the latest pip
version is installed in your working environment.
python3 -m pip install -U pip
Then you can install MotorNet using a simple pip install
command.
python3 -m pip install motornet
Requirements
There is no third-party software required to run MotorNet. However, some freely available python dependencies are required.
- TensorFlow: MotorNet is first and foremost built on TensorFlow. However, the standard
TensorFlow toolbox is not compatible with recent Apple machines equipped with M1 silicon chips, and users must rely on
an adapted version called tensorflow-macos. When installing MotorNet, the
setup.py
routine will automatically check the machine' OS platform and hardware to assess whether to solve for thetensorflow
ortensorflow-macos
dependency. - NumPy: For array and matrix computations when not using tensors.
- Matplotlib: For plotting utilities, mainly in the plotor module.
- IPython: Mainly for callbacks that output training metrics during model training.
- joblib: For parallelization routines in the parallelizer script.
Install with Anaconda
Installation via Anaconda is currently not supported.
Tutorials
There are several tutorials available to get you started, available in the repository's examples folder, as well as on the documentation website. Hopefully they will give a sense of how the API is supposed to work.
Changelog
See here for a curated log of update contents.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.