Code related to the paper "Using Physics Knowledge for Learning Rigid-body Forward Dynamics with Gaussian Process Force Priors" accepted at the Conference on Robot Learning (CoRL) 2021.
Main S-GP implementation can be found here: sgp/sgp.py.
Before starting the to train the models, it is necessary to generate the dataset, obtained from simulation. To do that, execute the script 01_simulate_KUKA_robot_arm.py, which requires, among other libraries, the pyBullet library installed.
This will execute the simulation and save the simulation results to KUKA-experiment/results/KUKA-surf-dataset/simdata_raw.dat. The name and location of the file, as well as many other simulation settings can be changed in the configuration file config_KUKA.py.
Next, execute the script 02_generate_dataset.py that generates a dataset file KUKA-experiment/results/KUKA-surf-dataset/simdata.dat. We are now ready to train the proposed learning methods.
In this experiment we compare the performance of a vanilla GP to the proposed Structured-GP w.r.t. to the number of training points used. Optionaly, an analytical model is used as a mean function to the S-GP. Execute the following main scripts to train the models
At the top of each script, you find the path to the model configuration file. Please, make sure you find the following path
cfg_model = importlib.import_module('results.KUKA-surf-dataset.exp_MAEvsTrainpoints.config_ML')which points to exp_MAEvsTrainpoints/config_ML.py.
Inside the config_ML file, there is an option s_gp.use_Fa_mean = False, that lets you change wheter the S-GP will use the analytical model or the zero mean as a mean function.
In addition, the option ds.datasetsize_train = 600 lets you change the number of traning points to be used by the different GP approaches during training.
Here, compare how the S-GP method performs when using a analytical multi-body dynamics model as a mean function, whose end-effector mass and CoG parameters are initially wrongly estimated. The S-GP approach aims at learning unmodeled friction forces alongside the analytical model parameters.
Execute the following main scripts to generate the results
and make sure the configuration file exp_learn_massCoG_alongside/config_ML.py is set in the first lines of each code:
cfg_model = importlib.import_module('results.KUKA-surf-dataset.exp_learn_massCoG_alongside.config_ML')
Execute the following main scripts to generate the results
and make sure each code points to the right configuration file exp_comp_gp-sgp-nn-mbd/config_ML.py:
cfg_model = importlib.import_module('results.KUKA-surf-dataset.exp_comp_gp-sgp-nn-mbd.config_ML')
This project is written 100% in python. The user is only required to install all python libraries used in this project, given in requirements.txt
Some useful links:
http://wiki.ros.org/urdfdom_py https://github.com/ros/urdf_parser_py
License: BSD
Author(s): Thomas Moulard, David Lu, Kelsey Hawkins, Antonio El Khoury, Eric Cousineau, Ioan Sucan , Jackie Kay
https://github.com/bulletphysics/bullet3/tree/master/data/kuka_iiwa https://github.com/bulletphysics/bullet3
License: Zlib