├── LICENSE ├── README.md ├── dexart ├── env │ ├── create_env.py │ ├── rl_env │ │ ├── base.py │ │ ├── bucket_env.py │ │ ├── faucet_env.py │ │ ├── laptop_env.py │ │ ├── pc_processing.py │ │ └── toilet_env.py │ ├── sim_env │ │ ├── base.py │ │ ├── bucket_env.py │ │ ├── constructor.py │ │ ├── faucet_env.py │ │ ├── laptop_env.py │ │ └── toilet_env.py │ └── task_setting.py └── utils │ ├── camera_utils.py │ ├── common_robot_utils.py │ ├── kinematics_helper.py │ ├── mesh_utils.py │ ├── model_utils.py │ ├── physical_scene_utils.py │ ├── random_utils.py │ └── render_scene_utils.py ├── docs ├── reproducibility.md └── teaser.png ├── examples ├── evaluate_policy.py ├── pretrain │ ├── generate_dataset.py │ ├── reconstruction │ │ ├── data_utils.py │ │ ├── models │ │ │ ├── pcn_util.py │ │ │ └── pointnet.py │ │ └── train_reconstruction.py │ ├── run.sh │ ├── segmentation │ │ ├── data_utils.py │ │ ├── models │ │ │ └── pointnet.py │ │ └── train_segmentation.py │ └── simsiam │ │ ├── data_utils.py │ │ ├── models │ │ └── pointnet.py │ │ └── train_simsiam.py ├── random_action.py ├── train.py ├── utils.py ├── visualize_observation.py └── visualize_policy.py ├── setup.py └── stable_baselines3 ├── __init__.py ├── a2c ├── __init__.py ├── a2c.py └── policies.py ├── common ├── __init__.py ├── base_class.py ├── buffers.py ├── callbacks.py ├── distributions.py ├── env_util.py ├── evaluation.py ├── logger.py ├── monitor.py ├── noise.py ├── on_policy_algorithm.py ├── policies.py ├── preprocessing.py ├── running_mean_std.py ├── save_util.py ├── torch_layers.py ├── type_aliases.py ├── utils.py └── vec_env │ ├── __init__.py │ ├── base_vec_env.py │ ├── dummy_vec_env.py │ ├── maniskill2_utils_common.py │ ├── maniskill2_utils_wrappers_obs.py │ ├── maniskill2_vec_env.py │ ├── maniskill2_wrapper_obs.py │ ├── stacked_observations.py │ ├── subproc_vec_env.py │ ├── util.py │ ├── vec_check_nan.py │ ├── vec_extract_dict_obs.py │ ├── vec_frame_stack.py │ ├── vec_monitor.py │ ├── vec_normalize.py │ ├── vec_transpose.py │ └── vec_video_recorder.py ├── networks └── pretrain_nets.py ├── pickle_utils.py ├── ppo ├── __init__.py ├── policies.py └── ppo.py ├── py.typed ├── simple_callback.py └── version.txt /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/README.md -------------------------------------------------------------------------------- /dexart/env/create_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/create_env.py -------------------------------------------------------------------------------- /dexart/env/rl_env/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/rl_env/base.py -------------------------------------------------------------------------------- /dexart/env/rl_env/bucket_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/rl_env/bucket_env.py -------------------------------------------------------------------------------- /dexart/env/rl_env/faucet_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/rl_env/faucet_env.py -------------------------------------------------------------------------------- /dexart/env/rl_env/laptop_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/rl_env/laptop_env.py -------------------------------------------------------------------------------- /dexart/env/rl_env/pc_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/rl_env/pc_processing.py -------------------------------------------------------------------------------- /dexart/env/rl_env/toilet_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/rl_env/toilet_env.py -------------------------------------------------------------------------------- /dexart/env/sim_env/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/sim_env/base.py -------------------------------------------------------------------------------- /dexart/env/sim_env/bucket_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/sim_env/bucket_env.py -------------------------------------------------------------------------------- /dexart/env/sim_env/constructor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/sim_env/constructor.py -------------------------------------------------------------------------------- /dexart/env/sim_env/faucet_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/sim_env/faucet_env.py -------------------------------------------------------------------------------- /dexart/env/sim_env/laptop_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/sim_env/laptop_env.py -------------------------------------------------------------------------------- /dexart/env/sim_env/toilet_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/sim_env/toilet_env.py -------------------------------------------------------------------------------- /dexart/env/task_setting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/env/task_setting.py -------------------------------------------------------------------------------- /dexart/utils/camera_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/utils/camera_utils.py -------------------------------------------------------------------------------- /dexart/utils/common_robot_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/utils/common_robot_utils.py -------------------------------------------------------------------------------- /dexart/utils/kinematics_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/utils/kinematics_helper.py -------------------------------------------------------------------------------- /dexart/utils/mesh_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/utils/mesh_utils.py -------------------------------------------------------------------------------- /dexart/utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/utils/model_utils.py -------------------------------------------------------------------------------- /dexart/utils/physical_scene_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/utils/physical_scene_utils.py -------------------------------------------------------------------------------- /dexart/utils/random_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/utils/random_utils.py -------------------------------------------------------------------------------- /dexart/utils/render_scene_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/dexart/utils/render_scene_utils.py -------------------------------------------------------------------------------- /docs/reproducibility.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/docs/reproducibility.md -------------------------------------------------------------------------------- /docs/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/docs/teaser.png -------------------------------------------------------------------------------- /examples/evaluate_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/evaluate_policy.py -------------------------------------------------------------------------------- /examples/pretrain/generate_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/pretrain/generate_dataset.py -------------------------------------------------------------------------------- /examples/pretrain/reconstruction/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/pretrain/reconstruction/data_utils.py -------------------------------------------------------------------------------- /examples/pretrain/reconstruction/models/pcn_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/pretrain/reconstruction/models/pcn_util.py -------------------------------------------------------------------------------- /examples/pretrain/reconstruction/models/pointnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/pretrain/reconstruction/models/pointnet.py -------------------------------------------------------------------------------- /examples/pretrain/reconstruction/train_reconstruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/pretrain/reconstruction/train_reconstruction.py -------------------------------------------------------------------------------- /examples/pretrain/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/pretrain/run.sh -------------------------------------------------------------------------------- /examples/pretrain/segmentation/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/pretrain/segmentation/data_utils.py -------------------------------------------------------------------------------- /examples/pretrain/segmentation/models/pointnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/pretrain/segmentation/models/pointnet.py -------------------------------------------------------------------------------- /examples/pretrain/segmentation/train_segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/pretrain/segmentation/train_segmentation.py -------------------------------------------------------------------------------- /examples/pretrain/simsiam/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/pretrain/simsiam/data_utils.py -------------------------------------------------------------------------------- /examples/pretrain/simsiam/models/pointnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/pretrain/simsiam/models/pointnet.py -------------------------------------------------------------------------------- /examples/pretrain/simsiam/train_simsiam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/pretrain/simsiam/train_simsiam.py -------------------------------------------------------------------------------- /examples/random_action.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/random_action.py -------------------------------------------------------------------------------- /examples/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/train.py -------------------------------------------------------------------------------- /examples/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/utils.py -------------------------------------------------------------------------------- /examples/visualize_observation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/visualize_observation.py -------------------------------------------------------------------------------- /examples/visualize_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/examples/visualize_policy.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/setup.py -------------------------------------------------------------------------------- /stable_baselines3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/__init__.py -------------------------------------------------------------------------------- /stable_baselines3/a2c/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/a2c/__init__.py -------------------------------------------------------------------------------- /stable_baselines3/a2c/a2c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/a2c/a2c.py -------------------------------------------------------------------------------- /stable_baselines3/a2c/policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/a2c/policies.py -------------------------------------------------------------------------------- /stable_baselines3/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /stable_baselines3/common/base_class.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/base_class.py -------------------------------------------------------------------------------- /stable_baselines3/common/buffers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/buffers.py -------------------------------------------------------------------------------- /stable_baselines3/common/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/callbacks.py -------------------------------------------------------------------------------- /stable_baselines3/common/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/distributions.py -------------------------------------------------------------------------------- /stable_baselines3/common/env_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/env_util.py -------------------------------------------------------------------------------- /stable_baselines3/common/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/evaluation.py -------------------------------------------------------------------------------- /stable_baselines3/common/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/logger.py -------------------------------------------------------------------------------- /stable_baselines3/common/monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/monitor.py -------------------------------------------------------------------------------- /stable_baselines3/common/noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/noise.py -------------------------------------------------------------------------------- /stable_baselines3/common/on_policy_algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/on_policy_algorithm.py -------------------------------------------------------------------------------- /stable_baselines3/common/policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/policies.py -------------------------------------------------------------------------------- /stable_baselines3/common/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/preprocessing.py -------------------------------------------------------------------------------- /stable_baselines3/common/running_mean_std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/running_mean_std.py -------------------------------------------------------------------------------- /stable_baselines3/common/save_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/save_util.py -------------------------------------------------------------------------------- /stable_baselines3/common/torch_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/torch_layers.py -------------------------------------------------------------------------------- /stable_baselines3/common/type_aliases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/type_aliases.py -------------------------------------------------------------------------------- /stable_baselines3/common/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/utils.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/__init__.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/base_vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/base_vec_env.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/dummy_vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/dummy_vec_env.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/maniskill2_utils_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/maniskill2_utils_common.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/maniskill2_utils_wrappers_obs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/maniskill2_utils_wrappers_obs.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/maniskill2_vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/maniskill2_vec_env.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/maniskill2_wrapper_obs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/maniskill2_wrapper_obs.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/stacked_observations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/stacked_observations.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/subproc_vec_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/subproc_vec_env.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/util.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/vec_check_nan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/vec_check_nan.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/vec_extract_dict_obs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/vec_extract_dict_obs.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/vec_frame_stack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/vec_frame_stack.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/vec_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/vec_monitor.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/vec_normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/vec_normalize.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/vec_transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/vec_transpose.py -------------------------------------------------------------------------------- /stable_baselines3/common/vec_env/vec_video_recorder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/common/vec_env/vec_video_recorder.py -------------------------------------------------------------------------------- /stable_baselines3/networks/pretrain_nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/networks/pretrain_nets.py -------------------------------------------------------------------------------- /stable_baselines3/pickle_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/pickle_utils.py -------------------------------------------------------------------------------- /stable_baselines3/ppo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/ppo/__init__.py -------------------------------------------------------------------------------- /stable_baselines3/ppo/policies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/ppo/policies.py -------------------------------------------------------------------------------- /stable_baselines3/ppo/ppo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/ppo/ppo.py -------------------------------------------------------------------------------- /stable_baselines3/py.typed: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /stable_baselines3/simple_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Kami-code/dexart-release/HEAD/stable_baselines3/simple_callback.py -------------------------------------------------------------------------------- /stable_baselines3/version.txt: -------------------------------------------------------------------------------- 1 | 1.5.1a6.dev 2 | --------------------------------------------------------------------------------