File size: 1,171 Bytes
c119e52 61de83e fab4866 61de83e e786752 61de83e 68fe22d 61de83e e1aeb25 61de83e e1aeb25 61de83e a0bb6c9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 | ---
license: bsd-2-clause
---
# Spatial attention recurrent neural network (SARNN)
The model was trained with the [MujocoUR5eCable dataset](https://huggingface.co/datasets/RoboManipBaselines/MujocoUR5eCable).
## Install
See [GitHub](https://github.com/isri-aist/RoboManipBaselines/blob/master/doc/install.md#SARNN) for installation.
## Policy rollout
Run a trained policy:
```console
# Go to the top directory of this repository
$ cd robo_manip_baselines
$ python ./bin/Rollout.py Sarnn MujocoUR5eCable --checkpoint ./checkpoint/Sarnn/<checkpoint_name>/policy_last.ckpt
```
## Technical Details
For more information on the technical details, please see the following paper:
```bib
@INPROCEEDINGS{SARNN_ICRA2022,
author = {Ichiwara, Hideyuki and Ito, Hiroshi and Yamamoto, Kenjiro and Mori, Hiroki and Ogata, Tetsuya},
title = {Contact-Rich Manipulation of a Flexible Object based on Deep Predictive Learning using Vision and Tactility},
booktitle = {International Conference on Robotics and Automation},
year = {2022},
pages = {5375-5381},
doi = {10.1109/ICRA46639.2022.9811940}
}
```
## RoboManipBaselines Paper
https://huggingface.co/papers/2509.17057
|