├── .gitignore ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── assets └── saved_models │ ├── mpo_state_rgb_test_triplet1 │ ├── saved_model.pb │ └── variables │ │ ├── variables.data-00000-of-00001 │ │ └── variables.index │ ├── mpo_state_rgb_test_triplet2 │ ├── saved_model.pb │ └── variables │ │ ├── variables.data-00000-of-00001 │ │ └── variables.index │ ├── mpo_state_rgb_test_triplet3 │ ├── saved_model.pb │ └── variables │ │ ├── variables.data-00000-of-00001 │ │ └── variables.index │ ├── mpo_state_rgb_test_triplet4 │ ├── saved_model.pb │ └── variables │ │ ├── variables.data-00000-of-00001 │ │ └── variables.index │ └── mpo_state_rgb_test_triplet5 │ ├── saved_model.pb │ └── variables │ ├── variables.data-00000-of-00001 │ └── variables.index ├── doc └── images │ └── rgb_environment.png ├── real_cell_documentation ├── basket_and_camera_assembly.pdf ├── bill_of_materials.pdf ├── cell_assembly.pdf ├── robot_assembly.pdf └── standard_cell_generic_setup.pdf ├── requirements.txt ├── rgb_stacking ├── environment.py ├── environment_test.py ├── main.py ├── physics_utils.py ├── reward_functions.py ├── stack_rewards.py ├── task.py └── utils │ ├── permissive_model.py │ └── policy_loading.py └── run.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/.gitignore -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/README.md -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet1/saved_model.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet1/saved_model.pb -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet1/variables/variables.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet1/variables/variables.data-00000-of-00001 -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet1/variables/variables.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet1/variables/variables.index -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet2/saved_model.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet2/saved_model.pb -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet2/variables/variables.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet2/variables/variables.data-00000-of-00001 -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet2/variables/variables.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet2/variables/variables.index -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet3/saved_model.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet3/saved_model.pb -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet3/variables/variables.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet3/variables/variables.data-00000-of-00001 -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet3/variables/variables.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet3/variables/variables.index -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet4/saved_model.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet4/saved_model.pb -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet4/variables/variables.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet4/variables/variables.data-00000-of-00001 -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet4/variables/variables.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet4/variables/variables.index -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet5/saved_model.pb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet5/saved_model.pb -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet5/variables/variables.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet5/variables/variables.data-00000-of-00001 -------------------------------------------------------------------------------- /assets/saved_models/mpo_state_rgb_test_triplet5/variables/variables.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/assets/saved_models/mpo_state_rgb_test_triplet5/variables/variables.index -------------------------------------------------------------------------------- /doc/images/rgb_environment.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/doc/images/rgb_environment.png -------------------------------------------------------------------------------- /real_cell_documentation/basket_and_camera_assembly.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/real_cell_documentation/basket_and_camera_assembly.pdf -------------------------------------------------------------------------------- /real_cell_documentation/bill_of_materials.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/real_cell_documentation/bill_of_materials.pdf -------------------------------------------------------------------------------- /real_cell_documentation/cell_assembly.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/real_cell_documentation/cell_assembly.pdf -------------------------------------------------------------------------------- /real_cell_documentation/robot_assembly.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/real_cell_documentation/robot_assembly.pdf -------------------------------------------------------------------------------- /real_cell_documentation/standard_cell_generic_setup.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/real_cell_documentation/standard_cell_generic_setup.pdf -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/requirements.txt -------------------------------------------------------------------------------- /rgb_stacking/environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/rgb_stacking/environment.py -------------------------------------------------------------------------------- /rgb_stacking/environment_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/rgb_stacking/environment_test.py -------------------------------------------------------------------------------- /rgb_stacking/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/rgb_stacking/main.py -------------------------------------------------------------------------------- /rgb_stacking/physics_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/rgb_stacking/physics_utils.py -------------------------------------------------------------------------------- /rgb_stacking/reward_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/rgb_stacking/reward_functions.py -------------------------------------------------------------------------------- /rgb_stacking/stack_rewards.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/rgb_stacking/stack_rewards.py -------------------------------------------------------------------------------- /rgb_stacking/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/rgb_stacking/task.py -------------------------------------------------------------------------------- /rgb_stacking/utils/permissive_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/rgb_stacking/utils/permissive_model.py -------------------------------------------------------------------------------- /rgb_stacking/utils/policy_loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/rgb_stacking/utils/policy_loading.py -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/google-deepmind/rgb_stacking/HEAD/run.sh --------------------------------------------------------------------------------