├── .gitignore ├── README.md ├── conda_environment.yaml ├── demo_push_data_collection.py ├── media └── teaser_image.png ├── pyrightconfig.json ├── setup.py ├── state_diff ├── codecs │ └── imagecodecs_numcodecs.py ├── common │ ├── checkpoint_util.py │ ├── cv2_util.py │ ├── env_util.py │ ├── json_logger.py │ ├── linear_interpolator.py │ ├── nested_dict_util.py │ ├── normalize_util.py │ ├── pose_trajectory_interpolator.py │ ├── precise_sleep.py │ ├── pymunk_override.py │ ├── pymunk_util.py │ ├── pytorch_util.py │ ├── replay_buffer.py │ ├── robomimic_config_util.py │ ├── robomimic_util.py │ ├── sampler.py │ └── timestamp_accumulator.py ├── config │ ├── task │ │ ├── blockpush_lowdim_seed.yaml │ │ ├── blockpush_lowdim_seed_abs.yaml │ │ ├── blockpush_traj_lowdim_seed.yaml │ │ ├── blockpush_traj_lowdim_seed_abs.yaml │ │ ├── kitchen_lowdim.yaml │ │ ├── kitchen_lowdim_abs.yaml │ │ ├── pushl_lowdim.yaml │ │ └── pushl_traj_lowdim.yaml │ ├── train_diffusion_trajectory_unet_lowdim_workspace.yaml │ └── train_diffusion_unet_lowdim_workspace.yaml ├── dataset │ ├── base_dataset.py │ ├── blockpush_lowdim_dataset.py │ ├── kitchen_lowdim_dataset.py │ ├── kitchen_mjl_lowdim_dataset.py │ ├── pusht_dataset.py │ └── pusht_traj_dataset.py ├── env │ ├── block_pushing │ │ ├── assets │ │ │ ├── block.urdf │ │ │ ├── block2.urdf │ │ │ ├── blocks │ │ │ │ ├── blue_cube.urdf │ │ │ │ ├── cube.obj │ │ │ │ ├── green_star.urdf │ │ │ │ ├── moon.obj │ │ │ │ ├── pentagon.obj │ │ │ │ ├── red_moon.urdf │ │ │ │ ├── star.obj │ │ │ │ └── yellow_pentagon.urdf │ │ │ ├── insert.urdf │ │ │ ├── plane.obj │ │ │ ├── suction │ │ │ │ ├── base.obj │ │ │ │ ├── cylinder.urdf │ │ │ │ ├── cylinder_real.urdf │ │ │ │ ├── head.obj │ │ │ │ ├── mid.obj │ │ │ │ ├── suction-base.urdf │ │ │ │ ├── suction-head-long.urdf │ │ │ │ ├── suction-head.urdf │ │ │ │ └── tip.obj │ │ │ ├── workspace.urdf │ │ │ ├── workspace_real.urdf │ │ │ ├── zone.obj │ │ │ ├── zone.urdf │ │ │ └── zone2.urdf │ │ ├── block_pushing.py │ │ ├── block_pushing_discontinuous.py │ │ ├── block_pushing_multimodal.py │ │ ├── oracles │ │ │ ├── discontinuous_push_oracle.py │ │ │ ├── multimodal_push_oracle.py │ │ │ ├── oriented_push_oracle.py │ │ │ ├── pushing_info.py │ │ │ └── reach_oracle.py │ │ └── utils │ │ │ ├── pose3d.py │ │ │ ├── utils_pybullet.py │ │ │ └── xarm_sim_robot.py │ ├── kitchen │ │ ├── __init__.py │ │ ├── base.py │ │ ├── kitchen_lowdim_wrapper.py │ │ ├── kitchen_util.py │ │ ├── relay_policy_learning │ │ │ ├── adept_envs │ │ │ │ ├── .pylintrc │ │ │ │ ├── .style.yapf │ │ │ │ └── adept_envs │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── base_robot.py │ │ │ │ │ ├── franka │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── assets │ │ │ │ │ │ └── franka_kitchen_jntpos_act_ab.xml │ │ │ │ │ ├── kitchen_multitask_v0.py │ │ │ │ │ └── robot │ │ │ │ │ │ ├── franka_config.xml │ │ │ │ │ │ └── franka_robot.py │ │ │ │ │ ├── mujoco_env.py │ │ │ │ │ ├── robot_env.py │ │ │ │ │ ├── simulation │ │ │ │ │ ├── module.py │ │ │ │ │ ├── renderer.py │ │ │ │ │ └── sim_robot.py │ │ │ │ │ └── utils │ │ │ │ │ ├── config.py │ │ │ │ │ ├── configurable.py │ │ │ │ │ ├── constants.py │ │ │ │ │ ├── parse_demos.py │ │ │ │ │ └── quatmath.py │ │ │ ├── adept_models │ │ │ │ ├── .gitignore │ │ │ │ ├── CONTRIBUTING.public.md │ │ │ │ ├── LICENSE │ │ │ │ ├── README.public.md │ │ │ │ ├── __init__.py │ │ │ │ ├── kitchen │ │ │ │ │ ├── assets │ │ │ │ │ │ ├── backwall_asset.xml │ │ │ │ │ │ ├── backwall_chain.xml │ │ │ │ │ │ ├── counters_asset.xml │ │ │ │ │ │ ├── counters_chain.xml │ │ │ │ │ │ ├── hingecabinet_asset.xml │ │ │ │ │ │ ├── hingecabinet_chain.xml │ │ │ │ │ │ ├── kettle_asset.xml │ │ │ │ │ │ ├── kettle_chain.xml │ │ │ │ │ │ ├── microwave_asset.xml │ │ │ │ │ │ ├── microwave_chain.xml │ │ │ │ │ │ ├── oven_asset.xml │ │ │ │ │ │ ├── oven_chain.xml │ │ │ │ │ │ ├── slidecabinet_asset.xml │ │ │ │ │ │ └── slidecabinet_chain.xml │ │ │ │ │ ├── counters.xml │ │ │ │ │ ├── hingecabinet.xml │ │ │ │ │ ├── kettle.xml │ │ │ │ │ ├── kitchen.xml │ │ │ │ │ ├── meshes │ │ │ │ │ │ ├── burnerplate.stl │ │ │ │ │ │ ├── burnerplate_mesh.stl │ │ │ │ │ │ ├── cabinetbase.stl │ │ │ │ │ │ ├── cabinetdrawer.stl │ │ │ │ │ │ ├── cabinethandle.stl │ │ │ │ │ │ ├── countertop.stl │ │ │ │ │ │ ├── faucet.stl │ │ │ │ │ │ ├── handle2.stl │ │ │ │ │ │ ├── hingecabinet.stl │ │ │ │ │ │ ├── hingedoor.stl │ │ │ │ │ │ ├── hingehandle.stl │ │ │ │ │ │ ├── hood.stl │ │ │ │ │ │ ├── kettle.stl │ │ │ │ │ │ ├── kettlehandle.stl │ │ │ │ │ │ ├── knob.stl │ │ │ │ │ │ ├── lightswitch.stl │ │ │ │ │ │ ├── lightswitchbase.stl │ │ │ │ │ │ ├── micro.stl │ │ │ │ │ │ ├── microbutton.stl │ │ │ │ │ │ ├── microdoor.stl │ │ │ │ │ │ ├── microefeet.stl │ │ │ │ │ │ ├── microfeet.stl │ │ │ │ │ │ ├── microhandle.stl │ │ │ │ │ │ ├── microwindow.stl │ │ │ │ │ │ ├── oven.stl │ │ │ │ │ │ ├── ovenhandle.stl │ │ │ │ │ │ ├── oventop.stl │ │ │ │ │ │ ├── ovenwindow.stl │ │ │ │ │ │ ├── slidecabinet.stl │ │ │ │ │ │ ├── slidedoor.stl │ │ │ │ │ │ ├── stoverim.stl │ │ │ │ │ │ ├── tile.stl │ │ │ │ │ │ └── wall.stl │ │ │ │ │ ├── microwave.xml │ │ │ │ │ ├── oven.xml │ │ │ │ │ ├── slidecabinet.xml │ │ │ │ │ └── textures │ │ │ │ │ │ ├── marble1.png │ │ │ │ │ │ ├── metal1.png │ │ │ │ │ │ ├── tile1.png │ │ │ │ │ │ └── wood1.png │ │ │ │ └── scenes │ │ │ │ │ ├── basic_scene.xml │ │ │ │ │ └── textures │ │ │ │ │ ├── white_marble_tile.png │ │ │ │ │ └── white_marble_tile2.png │ │ │ └── third_party │ │ │ │ └── franka │ │ │ │ ├── LICENSE │ │ │ │ ├── README.md │ │ │ │ ├── assets │ │ │ │ ├── actuator0.xml │ │ │ │ ├── actuator1.xml │ │ │ │ ├── assets.xml │ │ │ │ ├── basic_scene.xml │ │ │ │ ├── chain0.xml │ │ │ │ ├── chain0_overlay.xml │ │ │ │ ├── chain1.xml │ │ │ │ └── teleop_actuator.xml │ │ │ │ ├── bi-franka_panda.xml │ │ │ │ ├── franka_panda.png │ │ │ │ ├── franka_panda.xml │ │ │ │ ├── franka_panda_teleop.xml │ │ │ │ └── meshes │ │ │ │ ├── collision │ │ │ │ ├── finger.stl │ │ │ │ ├── hand.stl │ │ │ │ ├── link0.stl │ │ │ │ ├── link1.stl │ │ │ │ ├── link2.stl │ │ │ │ ├── link3.stl │ │ │ │ ├── link4.stl │ │ │ │ ├── link5.stl │ │ │ │ ├── link6.stl │ │ │ │ └── link7.stl │ │ │ │ └── visual │ │ │ │ ├── finger.stl │ │ │ │ ├── hand.stl │ │ │ │ ├── link0.stl │ │ │ │ ├── link1.stl │ │ │ │ ├── link2.stl │ │ │ │ ├── link3.stl │ │ │ │ ├── link4.stl │ │ │ │ ├── link5.stl │ │ │ │ ├── link6.stl │ │ │ │ └── link7.stl │ │ └── v0.py │ └── pusht │ │ ├── __init__.py │ │ ├── pushl_env.py │ │ ├── pushl_keypoints_env.py │ │ ├── pymunk_keypoint_manager.py │ │ └── pymunk_override.py ├── env_runner │ ├── base_lowdim_runner.py │ ├── blockpush_lowdim_runner.py │ ├── kitchen_lowdim_runner.py │ └── push_keypoints_runner.py ├── gym_util │ ├── async_vector_env.py │ ├── multistep_wrapper.py │ ├── sync_vector_env.py │ ├── video_recorder.py │ ├── video_recording_wrapper.py │ └── video_wrapper.py ├── model │ ├── common │ │ ├── dict_of_tensor_mixin.py │ │ ├── losses.py │ │ ├── lr_scheduler.py │ │ ├── module_attr_mixin.py │ │ ├── nn.py │ │ ├── normalizer.py │ │ ├── resample.py │ │ ├── rotation_transformer.py │ │ ├── shape_util.py │ │ └── tensor_util.py │ ├── diffusion │ │ ├── conditional_unet1d.py │ │ ├── conv1d_components.py │ │ ├── ema_model.py │ │ ├── mask_generator.py │ │ ├── positional_embedding.py │ │ └── transformer_for_diffusion.py │ └── vision │ │ ├── crop_randomizer.py │ │ └── model_getter.py ├── policy │ ├── base_lowdim_policy.py │ ├── diffusion_traj_unet_lowdim_policy.py │ └── diffusion_unet_lowdim_policy.py ├── scripts │ ├── bet_blockpush_conversion.py │ ├── blockpush_abs_conversion.py │ ├── episode_lengths.py │ ├── generate_bet_blockpush.py │ ├── real_dataset_conversion.py │ ├── real_pusht_metrics.py │ └── real_pusht_successrate.py ├── shared_memory │ ├── shared_memory_queue.py │ ├── shared_memory_ring_buffer.py │ ├── shared_memory_util.py │ └── shared_ndarray.py └── workspace │ ├── base_workspace.py │ └── train_diffusion_unet_lowdim_workspace.py ├── tests ├── test_block_pushing.py ├── test_cv2_util.py ├── test_multi_realsense.py ├── test_pose_trajectory_interpolator.py ├── test_precise_sleep.py ├── test_replay_buffer.py ├── test_ring_buffer.py ├── test_robomimic_image_runner.py ├── test_robomimic_lowdim_runner.py ├── test_shared_queue.py ├── test_single_realsense.py └── test_timestamp_accumulator.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/README.md -------------------------------------------------------------------------------- /conda_environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/conda_environment.yaml -------------------------------------------------------------------------------- /demo_push_data_collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/demo_push_data_collection.py -------------------------------------------------------------------------------- /media/teaser_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/media/teaser_image.png -------------------------------------------------------------------------------- /pyrightconfig.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/pyrightconfig.json -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/setup.py -------------------------------------------------------------------------------- /state_diff/codecs/imagecodecs_numcodecs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/codecs/imagecodecs_numcodecs.py -------------------------------------------------------------------------------- /state_diff/common/checkpoint_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/checkpoint_util.py -------------------------------------------------------------------------------- /state_diff/common/cv2_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/cv2_util.py -------------------------------------------------------------------------------- /state_diff/common/env_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/env_util.py -------------------------------------------------------------------------------- /state_diff/common/json_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/json_logger.py -------------------------------------------------------------------------------- /state_diff/common/linear_interpolator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/linear_interpolator.py -------------------------------------------------------------------------------- /state_diff/common/nested_dict_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/nested_dict_util.py -------------------------------------------------------------------------------- /state_diff/common/normalize_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/normalize_util.py -------------------------------------------------------------------------------- /state_diff/common/pose_trajectory_interpolator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/pose_trajectory_interpolator.py -------------------------------------------------------------------------------- /state_diff/common/precise_sleep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/precise_sleep.py -------------------------------------------------------------------------------- /state_diff/common/pymunk_override.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/pymunk_override.py -------------------------------------------------------------------------------- /state_diff/common/pymunk_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/pymunk_util.py -------------------------------------------------------------------------------- /state_diff/common/pytorch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/pytorch_util.py -------------------------------------------------------------------------------- /state_diff/common/replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/replay_buffer.py -------------------------------------------------------------------------------- /state_diff/common/robomimic_config_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/robomimic_config_util.py -------------------------------------------------------------------------------- /state_diff/common/robomimic_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/robomimic_util.py -------------------------------------------------------------------------------- /state_diff/common/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/sampler.py -------------------------------------------------------------------------------- /state_diff/common/timestamp_accumulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/common/timestamp_accumulator.py -------------------------------------------------------------------------------- /state_diff/config/task/blockpush_lowdim_seed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/config/task/blockpush_lowdim_seed.yaml -------------------------------------------------------------------------------- /state_diff/config/task/blockpush_lowdim_seed_abs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/config/task/blockpush_lowdim_seed_abs.yaml -------------------------------------------------------------------------------- /state_diff/config/task/blockpush_traj_lowdim_seed.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/config/task/blockpush_traj_lowdim_seed.yaml -------------------------------------------------------------------------------- /state_diff/config/task/blockpush_traj_lowdim_seed_abs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/config/task/blockpush_traj_lowdim_seed_abs.yaml -------------------------------------------------------------------------------- /state_diff/config/task/kitchen_lowdim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/config/task/kitchen_lowdim.yaml -------------------------------------------------------------------------------- /state_diff/config/task/kitchen_lowdim_abs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/config/task/kitchen_lowdim_abs.yaml -------------------------------------------------------------------------------- /state_diff/config/task/pushl_lowdim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/config/task/pushl_lowdim.yaml -------------------------------------------------------------------------------- /state_diff/config/task/pushl_traj_lowdim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/config/task/pushl_traj_lowdim.yaml -------------------------------------------------------------------------------- /state_diff/config/train_diffusion_trajectory_unet_lowdim_workspace.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/config/train_diffusion_trajectory_unet_lowdim_workspace.yaml -------------------------------------------------------------------------------- /state_diff/config/train_diffusion_unet_lowdim_workspace.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/config/train_diffusion_unet_lowdim_workspace.yaml -------------------------------------------------------------------------------- /state_diff/dataset/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/dataset/base_dataset.py -------------------------------------------------------------------------------- /state_diff/dataset/blockpush_lowdim_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/dataset/blockpush_lowdim_dataset.py -------------------------------------------------------------------------------- /state_diff/dataset/kitchen_lowdim_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/dataset/kitchen_lowdim_dataset.py -------------------------------------------------------------------------------- /state_diff/dataset/kitchen_mjl_lowdim_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/dataset/kitchen_mjl_lowdim_dataset.py -------------------------------------------------------------------------------- /state_diff/dataset/pusht_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/dataset/pusht_dataset.py -------------------------------------------------------------------------------- /state_diff/dataset/pusht_traj_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/dataset/pusht_traj_dataset.py -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/block.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/block.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/block2.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/block2.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/blocks/blue_cube.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/blocks/blue_cube.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/blocks/cube.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/blocks/cube.obj -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/blocks/green_star.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/blocks/green_star.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/blocks/moon.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/blocks/moon.obj -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/blocks/pentagon.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/blocks/pentagon.obj -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/blocks/red_moon.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/blocks/red_moon.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/blocks/star.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/blocks/star.obj -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/blocks/yellow_pentagon.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/blocks/yellow_pentagon.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/insert.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/insert.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/plane.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/plane.obj -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/suction/base.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/suction/base.obj -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/suction/cylinder.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/suction/cylinder.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/suction/cylinder_real.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/suction/cylinder_real.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/suction/head.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/suction/head.obj -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/suction/mid.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/suction/mid.obj -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/suction/suction-base.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/suction/suction-base.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/suction/suction-head-long.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/suction/suction-head-long.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/suction/suction-head.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/suction/suction-head.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/suction/tip.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/suction/tip.obj -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/workspace.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/workspace.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/workspace_real.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/workspace_real.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/zone.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/zone.obj -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/zone.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/zone.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/assets/zone2.urdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/assets/zone2.urdf -------------------------------------------------------------------------------- /state_diff/env/block_pushing/block_pushing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/block_pushing.py -------------------------------------------------------------------------------- /state_diff/env/block_pushing/block_pushing_discontinuous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/block_pushing_discontinuous.py -------------------------------------------------------------------------------- /state_diff/env/block_pushing/block_pushing_multimodal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/block_pushing_multimodal.py -------------------------------------------------------------------------------- /state_diff/env/block_pushing/oracles/discontinuous_push_oracle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/oracles/discontinuous_push_oracle.py -------------------------------------------------------------------------------- /state_diff/env/block_pushing/oracles/multimodal_push_oracle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/oracles/multimodal_push_oracle.py -------------------------------------------------------------------------------- /state_diff/env/block_pushing/oracles/oriented_push_oracle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/oracles/oriented_push_oracle.py -------------------------------------------------------------------------------- /state_diff/env/block_pushing/oracles/pushing_info.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/oracles/pushing_info.py -------------------------------------------------------------------------------- /state_diff/env/block_pushing/oracles/reach_oracle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/oracles/reach_oracle.py -------------------------------------------------------------------------------- /state_diff/env/block_pushing/utils/pose3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/utils/pose3d.py -------------------------------------------------------------------------------- /state_diff/env/block_pushing/utils/utils_pybullet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/utils/utils_pybullet.py -------------------------------------------------------------------------------- /state_diff/env/block_pushing/utils/xarm_sim_robot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/block_pushing/utils/xarm_sim_robot.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/__init__.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/base.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/kitchen_lowdim_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/kitchen_lowdim_wrapper.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/kitchen_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/kitchen_util.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/.pylintrc -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/.style.yapf -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/__init__.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/base_robot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/base_robot.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/franka/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/franka/__init__.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/franka/assets/franka_kitchen_jntpos_act_ab.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/franka/assets/franka_kitchen_jntpos_act_ab.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/franka/kitchen_multitask_v0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/franka/kitchen_multitask_v0.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/franka/robot/franka_config.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/franka/robot/franka_config.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/franka/robot/franka_robot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/franka/robot/franka_robot.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/mujoco_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/mujoco_env.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/robot_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/robot_env.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/simulation/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/simulation/module.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/simulation/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/simulation/renderer.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/simulation/sim_robot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/simulation/sim_robot.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/utils/config.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/utils/configurable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/utils/configurable.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/utils/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/utils/constants.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/utils/parse_demos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/utils/parse_demos.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/utils/quatmath.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_envs/adept_envs/utils/quatmath.py -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/.gitignore -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/CONTRIBUTING.public.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/CONTRIBUTING.public.md -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/LICENSE -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/README.public.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/README.public.md -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/backwall_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/backwall_asset.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/backwall_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/backwall_chain.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/counters_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/counters_asset.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/counters_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/counters_chain.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/hingecabinet_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/hingecabinet_asset.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/hingecabinet_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/hingecabinet_chain.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/kettle_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/kettle_asset.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/kettle_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/kettle_chain.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/microwave_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/microwave_asset.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/microwave_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/microwave_chain.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/oven_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/oven_asset.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/oven_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/oven_chain.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/slidecabinet_asset.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/slidecabinet_asset.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/slidecabinet_chain.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/assets/slidecabinet_chain.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/counters.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/counters.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/hingecabinet.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/hingecabinet.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/kettle.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/kettle.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/kitchen.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/kitchen.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/burnerplate.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/burnerplate.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/burnerplate_mesh.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/burnerplate_mesh.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/cabinetbase.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/cabinetbase.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/cabinetdrawer.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/cabinetdrawer.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/cabinethandle.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/cabinethandle.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/countertop.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/countertop.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/faucet.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/faucet.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/handle2.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/handle2.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/hingecabinet.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/hingecabinet.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/hingedoor.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/hingedoor.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/hingehandle.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/hingehandle.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/hood.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/hood.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/kettle.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/kettle.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/kettlehandle.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/kettlehandle.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/knob.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/knob.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/lightswitch.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/lightswitch.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/lightswitchbase.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/lightswitchbase.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/micro.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/micro.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/microbutton.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/microbutton.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/microdoor.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/microdoor.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/microefeet.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/microefeet.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/microfeet.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/microfeet.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/microhandle.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/microhandle.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/microwindow.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/microwindow.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/oven.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/oven.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/ovenhandle.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/ovenhandle.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/oventop.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/oventop.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/ovenwindow.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/ovenwindow.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/slidecabinet.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/slidecabinet.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/slidedoor.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/slidedoor.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/stoverim.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/stoverim.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/tile.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/tile.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/wall.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/meshes/wall.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/microwave.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/microwave.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/oven.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/oven.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/slidecabinet.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/slidecabinet.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/textures/marble1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/textures/marble1.png -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/textures/metal1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/textures/metal1.png -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/textures/tile1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/textures/tile1.png -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/textures/wood1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/kitchen/textures/wood1.png -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/scenes/basic_scene.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/scenes/basic_scene.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/scenes/textures/white_marble_tile.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/scenes/textures/white_marble_tile.png -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/adept_models/scenes/textures/white_marble_tile2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/adept_models/scenes/textures/white_marble_tile2.png -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/LICENSE -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/README.md -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/actuator0.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/actuator0.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/actuator1.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/actuator1.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/assets.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/assets.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/basic_scene.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/basic_scene.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/chain0.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/chain0.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/chain0_overlay.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/chain0_overlay.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/chain1.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/chain1.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/teleop_actuator.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/assets/teleop_actuator.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/bi-franka_panda.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/bi-franka_panda.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/franka_panda.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/franka_panda.png -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/franka_panda.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/franka_panda.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/franka_panda_teleop.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/franka_panda_teleop.xml -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/finger.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/finger.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/hand.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/hand.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link0.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link0.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link1.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link1.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link2.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link2.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link3.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link3.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link4.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link4.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link5.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link5.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link6.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link6.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link7.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/collision/link7.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/finger.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/finger.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/hand.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/hand.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link0.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link0.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link1.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link1.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link2.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link2.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link3.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link3.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link4.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link4.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link5.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link5.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link6.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link6.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link7.stl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/relay_policy_learning/third_party/franka/meshes/visual/link7.stl -------------------------------------------------------------------------------- /state_diff/env/kitchen/v0.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/kitchen/v0.py -------------------------------------------------------------------------------- /state_diff/env/pusht/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/pusht/__init__.py -------------------------------------------------------------------------------- /state_diff/env/pusht/pushl_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/pusht/pushl_env.py -------------------------------------------------------------------------------- /state_diff/env/pusht/pushl_keypoints_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/pusht/pushl_keypoints_env.py -------------------------------------------------------------------------------- /state_diff/env/pusht/pymunk_keypoint_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/pusht/pymunk_keypoint_manager.py -------------------------------------------------------------------------------- /state_diff/env/pusht/pymunk_override.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env/pusht/pymunk_override.py -------------------------------------------------------------------------------- /state_diff/env_runner/base_lowdim_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env_runner/base_lowdim_runner.py -------------------------------------------------------------------------------- /state_diff/env_runner/blockpush_lowdim_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env_runner/blockpush_lowdim_runner.py -------------------------------------------------------------------------------- /state_diff/env_runner/kitchen_lowdim_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env_runner/kitchen_lowdim_runner.py -------------------------------------------------------------------------------- /state_diff/env_runner/push_keypoints_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/env_runner/push_keypoints_runner.py -------------------------------------------------------------------------------- /state_diff/gym_util/async_vector_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/gym_util/async_vector_env.py -------------------------------------------------------------------------------- /state_diff/gym_util/multistep_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/gym_util/multistep_wrapper.py -------------------------------------------------------------------------------- /state_diff/gym_util/sync_vector_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/gym_util/sync_vector_env.py -------------------------------------------------------------------------------- /state_diff/gym_util/video_recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/gym_util/video_recorder.py -------------------------------------------------------------------------------- /state_diff/gym_util/video_recording_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/gym_util/video_recording_wrapper.py -------------------------------------------------------------------------------- /state_diff/gym_util/video_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/gym_util/video_wrapper.py -------------------------------------------------------------------------------- /state_diff/model/common/dict_of_tensor_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/common/dict_of_tensor_mixin.py -------------------------------------------------------------------------------- /state_diff/model/common/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/common/losses.py -------------------------------------------------------------------------------- /state_diff/model/common/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/common/lr_scheduler.py -------------------------------------------------------------------------------- /state_diff/model/common/module_attr_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/common/module_attr_mixin.py -------------------------------------------------------------------------------- /state_diff/model/common/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/common/nn.py -------------------------------------------------------------------------------- /state_diff/model/common/normalizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/common/normalizer.py -------------------------------------------------------------------------------- /state_diff/model/common/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/common/resample.py -------------------------------------------------------------------------------- /state_diff/model/common/rotation_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/common/rotation_transformer.py -------------------------------------------------------------------------------- /state_diff/model/common/shape_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/common/shape_util.py -------------------------------------------------------------------------------- /state_diff/model/common/tensor_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/common/tensor_util.py -------------------------------------------------------------------------------- /state_diff/model/diffusion/conditional_unet1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/diffusion/conditional_unet1d.py -------------------------------------------------------------------------------- /state_diff/model/diffusion/conv1d_components.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/diffusion/conv1d_components.py -------------------------------------------------------------------------------- /state_diff/model/diffusion/ema_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/diffusion/ema_model.py -------------------------------------------------------------------------------- /state_diff/model/diffusion/mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/diffusion/mask_generator.py -------------------------------------------------------------------------------- /state_diff/model/diffusion/positional_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/diffusion/positional_embedding.py -------------------------------------------------------------------------------- /state_diff/model/diffusion/transformer_for_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/diffusion/transformer_for_diffusion.py -------------------------------------------------------------------------------- /state_diff/model/vision/crop_randomizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/vision/crop_randomizer.py -------------------------------------------------------------------------------- /state_diff/model/vision/model_getter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/model/vision/model_getter.py -------------------------------------------------------------------------------- /state_diff/policy/base_lowdim_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/policy/base_lowdim_policy.py -------------------------------------------------------------------------------- /state_diff/policy/diffusion_traj_unet_lowdim_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/policy/diffusion_traj_unet_lowdim_policy.py -------------------------------------------------------------------------------- /state_diff/policy/diffusion_unet_lowdim_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/policy/diffusion_unet_lowdim_policy.py -------------------------------------------------------------------------------- /state_diff/scripts/bet_blockpush_conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/scripts/bet_blockpush_conversion.py -------------------------------------------------------------------------------- /state_diff/scripts/blockpush_abs_conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/scripts/blockpush_abs_conversion.py -------------------------------------------------------------------------------- /state_diff/scripts/episode_lengths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/scripts/episode_lengths.py -------------------------------------------------------------------------------- /state_diff/scripts/generate_bet_blockpush.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/scripts/generate_bet_blockpush.py -------------------------------------------------------------------------------- /state_diff/scripts/real_dataset_conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/scripts/real_dataset_conversion.py -------------------------------------------------------------------------------- /state_diff/scripts/real_pusht_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/scripts/real_pusht_metrics.py -------------------------------------------------------------------------------- /state_diff/scripts/real_pusht_successrate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/scripts/real_pusht_successrate.py -------------------------------------------------------------------------------- /state_diff/shared_memory/shared_memory_queue.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/shared_memory/shared_memory_queue.py -------------------------------------------------------------------------------- /state_diff/shared_memory/shared_memory_ring_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/shared_memory/shared_memory_ring_buffer.py -------------------------------------------------------------------------------- /state_diff/shared_memory/shared_memory_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/shared_memory/shared_memory_util.py -------------------------------------------------------------------------------- /state_diff/shared_memory/shared_ndarray.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/shared_memory/shared_ndarray.py -------------------------------------------------------------------------------- /state_diff/workspace/base_workspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/workspace/base_workspace.py -------------------------------------------------------------------------------- /state_diff/workspace/train_diffusion_unet_lowdim_workspace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/state_diff/workspace/train_diffusion_unet_lowdim_workspace.py -------------------------------------------------------------------------------- /tests/test_block_pushing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/tests/test_block_pushing.py -------------------------------------------------------------------------------- /tests/test_cv2_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/tests/test_cv2_util.py -------------------------------------------------------------------------------- /tests/test_multi_realsense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/tests/test_multi_realsense.py -------------------------------------------------------------------------------- /tests/test_pose_trajectory_interpolator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/tests/test_pose_trajectory_interpolator.py -------------------------------------------------------------------------------- /tests/test_precise_sleep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/tests/test_precise_sleep.py -------------------------------------------------------------------------------- /tests/test_replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/tests/test_replay_buffer.py -------------------------------------------------------------------------------- /tests/test_ring_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/tests/test_ring_buffer.py -------------------------------------------------------------------------------- /tests/test_robomimic_image_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/tests/test_robomimic_image_runner.py -------------------------------------------------------------------------------- /tests/test_robomimic_lowdim_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/tests/test_robomimic_lowdim_runner.py -------------------------------------------------------------------------------- /tests/test_shared_queue.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/tests/test_shared_queue.py -------------------------------------------------------------------------------- /tests/test_single_realsense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/tests/test_single_realsense.py -------------------------------------------------------------------------------- /tests/test_timestamp_accumulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/tests/test_timestamp_accumulator.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haonan16/state_diff/HEAD/train.py --------------------------------------------------------------------------------