By Daniel Seita et al. If you have questions on how use the code or the project in general, email me at seita@berkeley.edu.
Updates:
Here, you can find the paper, data, and videos.
This research was conducted at the AUTOLAB at UC Berkeley. http://autolab.berkeley.edu/
BibTex:
@inproceedings{seita_bedmake_2019,
author = {Daniel Seita and Nawid Jamali and Michael Laskey and Ajay Kumar Tanwani and Ron Berenstein and Prakash Baskaran and Soshi Iba and John Canny and Ken Goldberg},
title = {{Deep Transfer Learning of Pick Points on Fabric for Robot Bed-Making}},
booktitle = {International Symposium on Robotics Research (ISRR)},
Year = {2019}
}
Code:
Data:
If you unzip the data, you'll get stuff that looks like this for the grasp network and then the success network:
These are split into 10 pickle files for each network. Each of the 10 files (per neural network) is a standard pickle file (requires python 2.7 to load, not python 3, sorry, we were forced to use this due to code dependencies) which represents a standard python list. The lists have length equal to whatever is listed in the pickle file name. The data has already been shuffled and split into cross validation folds, which is what the `cv` represents in the file names. Of course, during training, the data needs to be further shuffled for minibatches. Each data point in the lists is a dictionary with `c_img` and `d_img` keys representing RGB and depth images. The latter is what we want to use. These also have labels within the dictionary.
Coverage Results:
VIDEO SUBMISSION
The video below is the main video we submitted.
OTHER VIDEOS
This shows a clip of a successful rollout with the teal blanket. Taken with an iPhone, so it's a bit shaky.
This shows one of the earlier failure cases with the offsets not being set correctly. This led us to adjust the offset of the gripper height as needed. In the future, we will also use a soft pad instead of a hard bed frame top.
Here's why the analytic baseline often gets good coverage. It doesn't grasp a corner but grasps a point such that the pull results in high coverage.