This repository is built based on SoftGym, with modified action space and added new testing environments and action space reduction method. The benchmarked tasks include
- cem_rope_straightening [source]
- cem_rope_folding [source]
- cem_cloth_diagonal_folding [source]
- cem_cloth_side_folding [source]
- cem_cloth_reflective_folding [source]
- cem_cloth_underneath_folding [source]
- Install SoftGym by following the instructions in SoftGym repository. Then, copy the softgym code to the SoftAgent root directory so we have the following file structure:
softagent ├── cem ├── ... ├── softgym
- Update conda env with additional packages required by SoftAgent:
conda env update --file environment.yml --prune
- Activate the conda environment by running
. ./prepare_1.0.sh
.
- Running rope straightening experiments:
cd cem_rope_straightening; python run_cem_rope_straightening.py
. Refer torun_cem_rope_straightening.py
for different arguments. - Running rope straightening experiments:
cd cem_rope_folding; python run_cem_rope_folding.py
. Refer torun_cem_rope_folding.py
for different arguments. - Running rope straightening experiments:
cd cem_cloth_diagonal_folding; python run_cem_cloth_diagonal_folding.py
. Refer torun_cem_cloth_diagonal_folding.py
for different arguments. - Running rope straightening experiments:
cd cem_cloth_side_folding; python run_cem_cloth_side_folding.py
. Refer torun_cem_cloth_side_folding.py
for different arguments. - Running rope straightening experiments:
cd cem_cloth_reflective_folding; python run_cem_cloth_reflective_folding.py
. Refer torun_cem_cloth_reflective_folding.py
for different arguments. - Running rope straightening experiments:
cd cem_cloth_underneath_folding; python run_cem_cloth_underneath_folding.py
. Refer torun_cem_cloth_underneath_folding.py
for different arguments.
If you find this codebase useful in your research, please consider citing:
@inproceedings{icra2023arsoftagent,
title={Goal-Conditioned Action Space Reduction for Deformable Object Manipulation},
author={Wang, Shengyin and Papallas, Rafael and Leonetti, Matteo and Dogar, Mehmet},
booktitle={IEEE International Conference on Robotics and Automation (ICRA)},
year={2023}
}
- softagent repository: https://github.com/Xingyu-Lin/softagent
- Softgym repository: https://github.com/Xingyu-Lin/softgym