├── .gitignore ├── LICENSE ├── README.md ├── composuite ├── __init__.py ├── arenas │ ├── compositional_arena.py │ ├── pick_place_arena.py │ ├── push_arena.py │ ├── shelf_arena.py │ └── trashcan_arena.py ├── assets │ ├── arenas │ │ ├── meshes │ │ │ ├── .DS_Store │ │ │ ├── trash_can.FCStd │ │ │ └── trash_can.stl │ │ ├── pick_place_arena.xml │ │ ├── push_arena.xml │ │ ├── shelf_arena.xml │ │ └── trashcan_arena.xml │ ├── objects │ │ ├── custombox-visual.xml │ │ ├── custombox.xml │ │ ├── dumbbell-visual.xml │ │ ├── dumbbell.xml │ │ ├── goal_wall.xml │ │ ├── hollowbox-visual.xml │ │ ├── hollowbox.xml │ │ ├── object_door_frame.xml │ │ ├── object_wall.xml │ │ ├── plate-visual.xml │ │ └── plate.xml │ └── textures │ │ ├── blue-wood.png │ │ ├── brass-ambra.png │ │ ├── bread.png │ │ ├── can.png │ │ ├── ceramic.png │ │ ├── cereal.png │ │ ├── clay.png │ │ ├── cream-plaster.png │ │ ├── dark-wood.png │ │ ├── dirt.png │ │ ├── glass.png │ │ ├── gray-felt.png │ │ ├── gray-plaster.png │ │ ├── gray-woodgrain.png │ │ ├── green-wood.png │ │ ├── lemon.png │ │ ├── light-wood.png │ │ ├── metal.png │ │ ├── pink-plaster.png │ │ ├── red-wood.png │ │ ├── steel-brushed.png │ │ ├── steel-scratched.png │ │ ├── white-bricks.png │ │ ├── white-plaster.png │ │ ├── wood-tiles.png │ │ ├── wood-varnished-panels.png │ │ └── yellow-plaster.png ├── demos │ ├── demo_compositional.py │ └── demo_compositional_teleop.py ├── env │ ├── compositional_env.py │ ├── gym_wrapper.py │ └── main.py ├── objects │ └── objects.py ├── tasks │ ├── pick_place_subtask.py │ ├── push_subtask.py │ ├── shelf_subtask.py │ └── trashcan_subtask.py └── utils │ ├── demo_utils.py │ └── mjk_utils.py ├── example_scripts └── train_ppo.py ├── requirements_default.txt ├── requirements_paper.txt ├── setup.py └── tests └── test_main_functions.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/README.md -------------------------------------------------------------------------------- /composuite/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/__init__.py -------------------------------------------------------------------------------- /composuite/arenas/compositional_arena.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/arenas/compositional_arena.py -------------------------------------------------------------------------------- /composuite/arenas/pick_place_arena.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/arenas/pick_place_arena.py -------------------------------------------------------------------------------- /composuite/arenas/push_arena.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/arenas/push_arena.py -------------------------------------------------------------------------------- /composuite/arenas/shelf_arena.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/arenas/shelf_arena.py -------------------------------------------------------------------------------- /composuite/arenas/trashcan_arena.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/arenas/trashcan_arena.py -------------------------------------------------------------------------------- /composuite/assets/arenas/meshes/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/arenas/meshes/.DS_Store -------------------------------------------------------------------------------- /composuite/assets/arenas/meshes/trash_can.FCStd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/arenas/meshes/trash_can.FCStd -------------------------------------------------------------------------------- /composuite/assets/arenas/meshes/trash_can.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/arenas/meshes/trash_can.stl -------------------------------------------------------------------------------- /composuite/assets/arenas/pick_place_arena.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/arenas/pick_place_arena.xml -------------------------------------------------------------------------------- /composuite/assets/arenas/push_arena.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/arenas/push_arena.xml -------------------------------------------------------------------------------- /composuite/assets/arenas/shelf_arena.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/arenas/shelf_arena.xml -------------------------------------------------------------------------------- /composuite/assets/arenas/trashcan_arena.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/arenas/trashcan_arena.xml -------------------------------------------------------------------------------- /composuite/assets/objects/custombox-visual.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/objects/custombox-visual.xml -------------------------------------------------------------------------------- /composuite/assets/objects/custombox.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/objects/custombox.xml -------------------------------------------------------------------------------- /composuite/assets/objects/dumbbell-visual.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/objects/dumbbell-visual.xml -------------------------------------------------------------------------------- /composuite/assets/objects/dumbbell.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/objects/dumbbell.xml -------------------------------------------------------------------------------- /composuite/assets/objects/goal_wall.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/objects/goal_wall.xml -------------------------------------------------------------------------------- /composuite/assets/objects/hollowbox-visual.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/objects/hollowbox-visual.xml -------------------------------------------------------------------------------- /composuite/assets/objects/hollowbox.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/objects/hollowbox.xml -------------------------------------------------------------------------------- /composuite/assets/objects/object_door_frame.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/objects/object_door_frame.xml -------------------------------------------------------------------------------- /composuite/assets/objects/object_wall.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/objects/object_wall.xml -------------------------------------------------------------------------------- /composuite/assets/objects/plate-visual.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/objects/plate-visual.xml -------------------------------------------------------------------------------- /composuite/assets/objects/plate.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/objects/plate.xml -------------------------------------------------------------------------------- /composuite/assets/textures/blue-wood.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/blue-wood.png -------------------------------------------------------------------------------- /composuite/assets/textures/brass-ambra.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/brass-ambra.png -------------------------------------------------------------------------------- /composuite/assets/textures/bread.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/bread.png -------------------------------------------------------------------------------- /composuite/assets/textures/can.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/can.png -------------------------------------------------------------------------------- /composuite/assets/textures/ceramic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/ceramic.png -------------------------------------------------------------------------------- /composuite/assets/textures/cereal.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/cereal.png -------------------------------------------------------------------------------- /composuite/assets/textures/clay.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/clay.png -------------------------------------------------------------------------------- /composuite/assets/textures/cream-plaster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/cream-plaster.png -------------------------------------------------------------------------------- /composuite/assets/textures/dark-wood.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/dark-wood.png -------------------------------------------------------------------------------- /composuite/assets/textures/dirt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/dirt.png -------------------------------------------------------------------------------- /composuite/assets/textures/glass.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/glass.png -------------------------------------------------------------------------------- /composuite/assets/textures/gray-felt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/gray-felt.png -------------------------------------------------------------------------------- /composuite/assets/textures/gray-plaster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/gray-plaster.png -------------------------------------------------------------------------------- /composuite/assets/textures/gray-woodgrain.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/gray-woodgrain.png -------------------------------------------------------------------------------- /composuite/assets/textures/green-wood.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/green-wood.png -------------------------------------------------------------------------------- /composuite/assets/textures/lemon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/lemon.png -------------------------------------------------------------------------------- /composuite/assets/textures/light-wood.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/light-wood.png -------------------------------------------------------------------------------- /composuite/assets/textures/metal.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/metal.png -------------------------------------------------------------------------------- /composuite/assets/textures/pink-plaster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/pink-plaster.png -------------------------------------------------------------------------------- /composuite/assets/textures/red-wood.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/red-wood.png -------------------------------------------------------------------------------- /composuite/assets/textures/steel-brushed.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/steel-brushed.png -------------------------------------------------------------------------------- /composuite/assets/textures/steel-scratched.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/steel-scratched.png -------------------------------------------------------------------------------- /composuite/assets/textures/white-bricks.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/white-bricks.png -------------------------------------------------------------------------------- /composuite/assets/textures/white-plaster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/white-plaster.png -------------------------------------------------------------------------------- /composuite/assets/textures/wood-tiles.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/wood-tiles.png -------------------------------------------------------------------------------- /composuite/assets/textures/wood-varnished-panels.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/wood-varnished-panels.png -------------------------------------------------------------------------------- /composuite/assets/textures/yellow-plaster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/assets/textures/yellow-plaster.png -------------------------------------------------------------------------------- /composuite/demos/demo_compositional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/demos/demo_compositional.py -------------------------------------------------------------------------------- /composuite/demos/demo_compositional_teleop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/demos/demo_compositional_teleop.py -------------------------------------------------------------------------------- /composuite/env/compositional_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/env/compositional_env.py -------------------------------------------------------------------------------- /composuite/env/gym_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/env/gym_wrapper.py -------------------------------------------------------------------------------- /composuite/env/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/env/main.py -------------------------------------------------------------------------------- /composuite/objects/objects.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/objects/objects.py -------------------------------------------------------------------------------- /composuite/tasks/pick_place_subtask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/tasks/pick_place_subtask.py -------------------------------------------------------------------------------- /composuite/tasks/push_subtask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/tasks/push_subtask.py -------------------------------------------------------------------------------- /composuite/tasks/shelf_subtask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/tasks/shelf_subtask.py -------------------------------------------------------------------------------- /composuite/tasks/trashcan_subtask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/tasks/trashcan_subtask.py -------------------------------------------------------------------------------- /composuite/utils/demo_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/utils/demo_utils.py -------------------------------------------------------------------------------- /composuite/utils/mjk_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/composuite/utils/mjk_utils.py -------------------------------------------------------------------------------- /example_scripts/train_ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/example_scripts/train_ppo.py -------------------------------------------------------------------------------- /requirements_default.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/requirements_default.txt -------------------------------------------------------------------------------- /requirements_paper.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/requirements_paper.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_main_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lifelong-ML/CompoSuite/HEAD/tests/test_main_functions.py --------------------------------------------------------------------------------