The following is a meta-repository containing all of the packages required to run code for the Contact/STS Imitation Learning project.
contact-il
: The contact_il package that runs most of the imitation learning code, including demonstration collection and testing, as well as config parameters to be used with learning code fromplace_from_pick_learning
.panda-polymetis
: An interface between Meta's polymetis and the environments used in this package.contact-panda-envs
: Gym environments for contact/tactile tasks to be used with a simulated or real Franka Panda.sts
: A library for processing sts data.pysts
: Sensor interfacing code that can be used to acquire STS data.transform_utils
: A set of generic utilities for working with SO3 and SE3 transformations.place_from_pick_learning
: Code based on another project used for training policies.
Installation instructions for these packages are all found in the contact-il
sub-package.
The training and testing datasets corresponding to this project are currently closed-source and owned by Samsung.