├── .gitignore ├── LICENSE ├── README.md ├── arm ├── __init__.py ├── arm │ ├── __init__.py │ ├── launch_utils.py │ ├── next_best_pose_agent.py │ └── qattention_agent.py ├── baselines │ ├── __init__.py │ ├── bc │ │ ├── __init__.py │ │ ├── bc_agent.py │ │ └── launch_utils.py │ ├── dac │ │ ├── __init__.py │ │ ├── dac_agent.py │ │ └── launch_utils.py │ ├── sac │ │ ├── __init__.py │ │ ├── launch_utils.py │ │ └── sac_agent.py │ └── td3 │ │ ├── __init__.py │ │ ├── launch_utils.py │ │ └── td3_agent.py ├── c2farm │ ├── __init__.py │ ├── launch_utils.py │ ├── networks.py │ ├── qattention_agent.py │ ├── qattention_stack_agent.py │ └── voxel_grid.py ├── custom_rlbench_env.py ├── demo_loading_utils.py ├── lpr │ ├── __init__.py │ ├── combined_agent.py │ ├── const.py │ ├── launch_utils.py │ ├── networks.py │ ├── rlbench_path_sampler.py │ ├── rollout_generator.py │ ├── trajectory_action_mode.py │ └── trajectory_agent.py ├── network_utils.py ├── preprocess_agent.py ├── qte │ ├── __init__.py │ ├── launch_utils.py │ ├── networks.py │ └── qattention_agent.py └── utils.py ├── conf ├── config.yaml └── method │ ├── ARM.yaml │ ├── BC.yaml │ ├── C2FARM.yaml │ ├── DAC.yaml │ ├── LPR.yaml │ ├── QTE.yaml │ ├── SAC.yaml │ └── TD3.yaml ├── launch.py ├── readme_files └── variants.png ├── requirements.txt └── tools ├── run_agent.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/README.md -------------------------------------------------------------------------------- /arm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /arm/arm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/arm/__init__.py -------------------------------------------------------------------------------- /arm/arm/launch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/arm/launch_utils.py -------------------------------------------------------------------------------- /arm/arm/next_best_pose_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/arm/next_best_pose_agent.py -------------------------------------------------------------------------------- /arm/arm/qattention_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/arm/qattention_agent.py -------------------------------------------------------------------------------- /arm/baselines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /arm/baselines/bc/__init__.py: -------------------------------------------------------------------------------- 1 | import arm.baselines.bc.launch_utils 2 | -------------------------------------------------------------------------------- /arm/baselines/bc/bc_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/baselines/bc/bc_agent.py -------------------------------------------------------------------------------- /arm/baselines/bc/launch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/baselines/bc/launch_utils.py -------------------------------------------------------------------------------- /arm/baselines/dac/__init__.py: -------------------------------------------------------------------------------- 1 | import arm.baselines.dac.launch_utils 2 | -------------------------------------------------------------------------------- /arm/baselines/dac/dac_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/baselines/dac/dac_agent.py -------------------------------------------------------------------------------- /arm/baselines/dac/launch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/baselines/dac/launch_utils.py -------------------------------------------------------------------------------- /arm/baselines/sac/__init__.py: -------------------------------------------------------------------------------- 1 | import arm.baselines.sac.launch_utils 2 | -------------------------------------------------------------------------------- /arm/baselines/sac/launch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/baselines/sac/launch_utils.py -------------------------------------------------------------------------------- /arm/baselines/sac/sac_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/baselines/sac/sac_agent.py -------------------------------------------------------------------------------- /arm/baselines/td3/__init__.py: -------------------------------------------------------------------------------- 1 | import arm.baselines.td3.launch_utils 2 | -------------------------------------------------------------------------------- /arm/baselines/td3/launch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/baselines/td3/launch_utils.py -------------------------------------------------------------------------------- /arm/baselines/td3/td3_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/baselines/td3/td3_agent.py -------------------------------------------------------------------------------- /arm/c2farm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/c2farm/__init__.py -------------------------------------------------------------------------------- /arm/c2farm/launch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/c2farm/launch_utils.py -------------------------------------------------------------------------------- /arm/c2farm/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/c2farm/networks.py -------------------------------------------------------------------------------- /arm/c2farm/qattention_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/c2farm/qattention_agent.py -------------------------------------------------------------------------------- /arm/c2farm/qattention_stack_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/c2farm/qattention_stack_agent.py -------------------------------------------------------------------------------- /arm/c2farm/voxel_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/c2farm/voxel_grid.py -------------------------------------------------------------------------------- /arm/custom_rlbench_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/custom_rlbench_env.py -------------------------------------------------------------------------------- /arm/demo_loading_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/demo_loading_utils.py -------------------------------------------------------------------------------- /arm/lpr/__init__.py: -------------------------------------------------------------------------------- 1 | import arm.lpr.launch_utils 2 | -------------------------------------------------------------------------------- /arm/lpr/combined_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/lpr/combined_agent.py -------------------------------------------------------------------------------- /arm/lpr/const.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/lpr/const.py -------------------------------------------------------------------------------- /arm/lpr/launch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/lpr/launch_utils.py -------------------------------------------------------------------------------- /arm/lpr/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/lpr/networks.py -------------------------------------------------------------------------------- /arm/lpr/rlbench_path_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/lpr/rlbench_path_sampler.py -------------------------------------------------------------------------------- /arm/lpr/rollout_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/lpr/rollout_generator.py -------------------------------------------------------------------------------- /arm/lpr/trajectory_action_mode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/lpr/trajectory_action_mode.py -------------------------------------------------------------------------------- /arm/lpr/trajectory_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/lpr/trajectory_agent.py -------------------------------------------------------------------------------- /arm/network_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/network_utils.py -------------------------------------------------------------------------------- /arm/preprocess_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/preprocess_agent.py -------------------------------------------------------------------------------- /arm/qte/__init__.py: -------------------------------------------------------------------------------- 1 | import arm.qte.launch_utils 2 | -------------------------------------------------------------------------------- /arm/qte/launch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/qte/launch_utils.py -------------------------------------------------------------------------------- /arm/qte/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/qte/networks.py -------------------------------------------------------------------------------- /arm/qte/qattention_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/qte/qattention_agent.py -------------------------------------------------------------------------------- /arm/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/arm/utils.py -------------------------------------------------------------------------------- /conf/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/conf/config.yaml -------------------------------------------------------------------------------- /conf/method/ARM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/conf/method/ARM.yaml -------------------------------------------------------------------------------- /conf/method/BC.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/conf/method/BC.yaml -------------------------------------------------------------------------------- /conf/method/C2FARM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/conf/method/C2FARM.yaml -------------------------------------------------------------------------------- /conf/method/DAC.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/conf/method/DAC.yaml -------------------------------------------------------------------------------- /conf/method/LPR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/conf/method/LPR.yaml -------------------------------------------------------------------------------- /conf/method/QTE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/conf/method/QTE.yaml -------------------------------------------------------------------------------- /conf/method/SAC.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/conf/method/SAC.yaml -------------------------------------------------------------------------------- /conf/method/TD3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/conf/method/TD3.yaml -------------------------------------------------------------------------------- /launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/launch.py -------------------------------------------------------------------------------- /readme_files/variants.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/readme_files/variants.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | hydra-core 2 | scipy 3 | shapely 4 | trimesh 5 | pyrender 6 | -------------------------------------------------------------------------------- /tools/run_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/tools/run_agent.py -------------------------------------------------------------------------------- /tools/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/stepjam/ARM/HEAD/tools/utils.py --------------------------------------------------------------------------------