ykitagawa commited on
Commit
61de83e
·
verified ·
1 Parent(s): e1aeb25

fixed README

Browse files
Files changed (1) hide show
  1. README.md +24 -49
README.md CHANGED
@@ -1,51 +1,26 @@
1
- ---
2
- license: bsd-2-clause
3
- ---
4
- # [RoboManipBaselines](https://github.com/isri-aist/RoboManipBaselines)
5
- Software that integrates various imitation learning methods and benchmark task environments to provide baselines for robot manipulation
6
-
7
- [![CI-install](https://github.com/isri-aist/RoboManipBaselines/actions/workflows/install.yml/badge.svg)](https://github.com/isri-aist/RoboManipBaselines/actions/workflows/install.yml)
8
- [![CI-pre-commit](https://github.com/isri-aist/RoboManipBaselines/actions/workflows/pre-commit.yml/badge.svg)](https://github.com/isri-aist/RoboManipBaselines/actions/workflows/pre-commit.yml)
9
- [![LICENSE](https://img.shields.io/github/license/isri-aist/RoboManipBaselines)](https://github.com/isri-aist/RoboManipBaselines/blob/master/LICENSE)
10
-
11
- https://github.com/user-attachments/assets/ba4a772f-0de5-47da-a4ec-bdcbf13d7d58
12
-
13
- ## Policies
14
- ### [SARNN](./robo_manip_baselines/sarnn)
15
- Spatial attention recurrent neural network
16
-
17
- ### [ACT](./robo_manip_baselines/act)
18
- Action Chunking with Transformers
19
-
20
- ### [DiffusionPolicy](./robo_manip_baselines/diffusion_policy)
21
- Diffusion Policy
22
-
23
- ### [MT-ACT](./robo_manip_baselines/mt_act)
24
- Multi-Task Action Chunking Transformer
25
-
26
- ## Data
27
- ### Publicly available datasets
28
-
29
- ### Data collection by teleoperation
30
-
31
- ## Environments for robot manipulation
32
-
33
- ## Utilities
34
-
35
- ## Evaluation results
36
-
37
- ## License
38
- Files that originate from this repository are subject to the BSD 2-Clause License. If a file explicitly states a different license, or if there are different license files in a directory, those licenses will take precedence. For files in third-party directories, please follow the respective licenses.
39
-
40
- ## Citation
41
- You can cite this work with:
42
  ```bib
43
- @software{RoboManipBaselines_GitHub2024,
44
- author = {Murooka, Masaki and Motoda, Tomohiro and Nakajo, Ryoichi},
45
- title = {{RoboManipBaselines}},
46
- url = {https://github.com/isri-aist/RoboManipBaselines},
47
- version = {1.0.0},
48
- year = {2024}
49
- month = dec,
50
  }
51
- ```
 
1
+ # Spatial attention recurrent neural network (SARNN)
2
+
3
+ ## Install
4
+ See [GitHub](https://github.com/isri-aist/RoboManipBaselines/tree/master/robo_manip_baselines/sarnn) for installation.
5
+
6
+ ## Policy rollout
7
+ Run a trained policy:
8
+ ```console
9
+ $ python ./bin/rollout/RolloutSarnnMujocoUR5eCable.py \
10
+ --checkpoint <checkpoint_path>/SARNN.pth \
11
+ --cropped_img_size 280 --skip 6 --world_idx 0
12
+ ```
13
+ The `--cropped_img_size` option must be the same as for dataset generation.
14
+
15
+ ## Technical Details
16
+ For more information on the technical details, please see the following paper:
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  ```bib
18
+ @INPROCEEDINGS{SARNN_ICRA2022,
19
+ author = {Ichiwara, Hideyuki and Ito, Hiroshi and Yamamoto, Kenjiro and Mori, Hiroki and Ogata, Tetsuya},
20
+ title = {Contact-Rich Manipulation of a Flexible Object based on Deep Predictive Learning using Vision and Tactility},
21
+ booktitle = {International Conference on Robotics and Automation},
22
+ year = {2022},
23
+ pages = {5375-5381},
24
+ doi = {10.1109/ICRA46639.2022.9811940}
25
  }
26
+ ```