├── .gitignore ├── LICENSE ├── README.md ├── arm_controller.py ├── arm_server.py ├── base_controller.py ├── base_server.py ├── cameras.py ├── constants.py ├── convert_to_robomimic_hdf5.py ├── diffusion-policy.patch ├── episode_storage.py ├── gamepad_teleop.py ├── ik_solver.py ├── kinova.py ├── main.py ├── models ├── assets │ ├── 2f85 │ │ ├── base.stl │ │ ├── coupler.stl │ │ ├── driver.stl │ │ ├── follower.stl │ │ ├── pad.stl │ │ ├── silicone_pad.stl │ │ └── spring_link.stl │ ├── base │ │ ├── arm_plate.stl │ │ ├── body.stl │ │ ├── bottom_plate.stl │ │ ├── bumper.stl │ │ └── top_plate.stl │ └── gen3 │ │ ├── base_link.stl │ │ ├── bracelet_no_vision_link.stl │ │ ├── bracelet_with_vision_link.stl │ │ ├── forearm_link.stl │ │ ├── half_arm_1_link.stl │ │ ├── half_arm_2_link.stl │ │ ├── shoulder_link.stl │ │ ├── spherical_wrist_1_link.stl │ │ └── spherical_wrist_2_link.stl ├── gen3_robotiq_2f_85.urdf ├── kinova_gen3 │ ├── gen3.xml │ ├── gen3_2f85.xml │ ├── scene.xml │ └── scene_2f85.xml └── stanford_tidybot │ ├── base.xml │ ├── scene.xml │ ├── scene_base.xml │ └── tidybot.xml ├── mujoco_env.py ├── plot_base_state.py ├── plot_demos.ipynb ├── policies.py ├── policy_server.py ├── real_env.py ├── replay_episodes.py ├── requirements.txt ├── static └── js │ └── webxr-button.js ├── templates └── index.html └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/README.md -------------------------------------------------------------------------------- /arm_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/arm_controller.py -------------------------------------------------------------------------------- /arm_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/arm_server.py -------------------------------------------------------------------------------- /base_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/base_controller.py -------------------------------------------------------------------------------- /base_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/base_server.py -------------------------------------------------------------------------------- /cameras.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/cameras.py -------------------------------------------------------------------------------- /constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/constants.py -------------------------------------------------------------------------------- /convert_to_robomimic_hdf5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/convert_to_robomimic_hdf5.py -------------------------------------------------------------------------------- /diffusion-policy.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/diffusion-policy.patch -------------------------------------------------------------------------------- /episode_storage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/episode_storage.py -------------------------------------------------------------------------------- /gamepad_teleop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/gamepad_teleop.py -------------------------------------------------------------------------------- /ik_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/ik_solver.py -------------------------------------------------------------------------------- /kinova.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/kinova.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/main.py -------------------------------------------------------------------------------- /models/assets/2f85/base.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/2f85/base.stl -------------------------------------------------------------------------------- /models/assets/2f85/coupler.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/2f85/coupler.stl -------------------------------------------------------------------------------- /models/assets/2f85/driver.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/2f85/driver.stl -------------------------------------------------------------------------------- /models/assets/2f85/follower.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/2f85/follower.stl -------------------------------------------------------------------------------- /models/assets/2f85/pad.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/2f85/pad.stl -------------------------------------------------------------------------------- /models/assets/2f85/silicone_pad.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/2f85/silicone_pad.stl -------------------------------------------------------------------------------- /models/assets/2f85/spring_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/2f85/spring_link.stl -------------------------------------------------------------------------------- /models/assets/base/arm_plate.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/base/arm_plate.stl -------------------------------------------------------------------------------- /models/assets/base/body.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/base/body.stl -------------------------------------------------------------------------------- /models/assets/base/bottom_plate.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/base/bottom_plate.stl -------------------------------------------------------------------------------- /models/assets/base/bumper.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/base/bumper.stl -------------------------------------------------------------------------------- /models/assets/base/top_plate.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/base/top_plate.stl -------------------------------------------------------------------------------- /models/assets/gen3/base_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/gen3/base_link.stl -------------------------------------------------------------------------------- /models/assets/gen3/bracelet_no_vision_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/gen3/bracelet_no_vision_link.stl -------------------------------------------------------------------------------- /models/assets/gen3/bracelet_with_vision_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/gen3/bracelet_with_vision_link.stl -------------------------------------------------------------------------------- /models/assets/gen3/forearm_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/gen3/forearm_link.stl -------------------------------------------------------------------------------- /models/assets/gen3/half_arm_1_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/gen3/half_arm_1_link.stl -------------------------------------------------------------------------------- /models/assets/gen3/half_arm_2_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/gen3/half_arm_2_link.stl -------------------------------------------------------------------------------- /models/assets/gen3/shoulder_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/gen3/shoulder_link.stl -------------------------------------------------------------------------------- /models/assets/gen3/spherical_wrist_1_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/gen3/spherical_wrist_1_link.stl -------------------------------------------------------------------------------- /models/assets/gen3/spherical_wrist_2_link.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/assets/gen3/spherical_wrist_2_link.stl -------------------------------------------------------------------------------- /models/gen3_robotiq_2f_85.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/gen3_robotiq_2f_85.urdf -------------------------------------------------------------------------------- /models/kinova_gen3/gen3.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/kinova_gen3/gen3.xml -------------------------------------------------------------------------------- /models/kinova_gen3/gen3_2f85.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/kinova_gen3/gen3_2f85.xml -------------------------------------------------------------------------------- /models/kinova_gen3/scene.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/kinova_gen3/scene.xml -------------------------------------------------------------------------------- /models/kinova_gen3/scene_2f85.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/kinova_gen3/scene_2f85.xml -------------------------------------------------------------------------------- /models/stanford_tidybot/base.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/stanford_tidybot/base.xml -------------------------------------------------------------------------------- /models/stanford_tidybot/scene.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/stanford_tidybot/scene.xml -------------------------------------------------------------------------------- /models/stanford_tidybot/scene_base.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/stanford_tidybot/scene_base.xml -------------------------------------------------------------------------------- /models/stanford_tidybot/tidybot.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/models/stanford_tidybot/tidybot.xml -------------------------------------------------------------------------------- /mujoco_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/mujoco_env.py -------------------------------------------------------------------------------- /plot_base_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/plot_base_state.py -------------------------------------------------------------------------------- /plot_demos.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/plot_demos.ipynb -------------------------------------------------------------------------------- /policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/policies.py -------------------------------------------------------------------------------- /policy_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/policy_server.py -------------------------------------------------------------------------------- /real_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/real_env.py -------------------------------------------------------------------------------- /replay_episodes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/replay_episodes.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/requirements.txt -------------------------------------------------------------------------------- /static/js/webxr-button.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/static/js/webxr-button.js -------------------------------------------------------------------------------- /templates/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/templates/index.html -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jimmyyhwu/tidybot2/HEAD/utils.py --------------------------------------------------------------------------------