├── LICENSE ├── README.md ├── __init__.py ├── agents ├── __init__.py ├── agents_3d │ ├── __init__.py │ ├── base_3d.py │ └── dqn_3d_asr.py └── base_agent.py ├── images ├── UR5_setup.png ├── dualbin_paralle_run.png ├── simluation_obs.png ├── simulation_env.png └── simulation_training_obj_set.png ├── networks ├── __init__.py └── equivariant_models_refactor.py ├── requirements.txt ├── scripts ├── __init__.py ├── create_agent.py ├── main.py ├── plot_multiple.py ├── test_robot.py ├── train_robot.py └── train_robot_parallel.py ├── storage ├── __init__.py └── buffer.py └── utils ├── __init__.py ├── env_wrapper.py ├── logger.py ├── parallel_utils.py ├── parameters.py ├── torch_utils.py ├── transformations.py └── visualization_utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /agents/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /agents/agents_3d/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /agents/agents_3d/base_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/agents/agents_3d/base_3d.py -------------------------------------------------------------------------------- /agents/agents_3d/dqn_3d_asr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/agents/agents_3d/dqn_3d_asr.py -------------------------------------------------------------------------------- /agents/base_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/agents/base_agent.py -------------------------------------------------------------------------------- /images/UR5_setup.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/images/UR5_setup.png -------------------------------------------------------------------------------- /images/dualbin_paralle_run.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/images/dualbin_paralle_run.png -------------------------------------------------------------------------------- /images/simluation_obs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/images/simluation_obs.png -------------------------------------------------------------------------------- /images/simulation_env.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/images/simulation_env.png -------------------------------------------------------------------------------- /images/simulation_training_obj_set.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/images/simulation_training_obj_set.png -------------------------------------------------------------------------------- /networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/networks/__init__.py -------------------------------------------------------------------------------- /networks/equivariant_models_refactor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/networks/equivariant_models_refactor.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/create_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/scripts/create_agent.py -------------------------------------------------------------------------------- /scripts/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/scripts/main.py -------------------------------------------------------------------------------- /scripts/plot_multiple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/scripts/plot_multiple.py -------------------------------------------------------------------------------- /scripts/test_robot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/scripts/test_robot.py -------------------------------------------------------------------------------- /scripts/train_robot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/scripts/train_robot.py -------------------------------------------------------------------------------- /scripts/train_robot_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/scripts/train_robot_parallel.py -------------------------------------------------------------------------------- /storage/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /storage/buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/storage/buffer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/env_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/utils/env_wrapper.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/parallel_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/utils/parallel_utils.py -------------------------------------------------------------------------------- /utils/parameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/utils/parameters.py -------------------------------------------------------------------------------- /utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/utils/torch_utils.py -------------------------------------------------------------------------------- /utils/transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/utils/transformations.py -------------------------------------------------------------------------------- /utils/visualization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZXP-S-works/SE2-equivariant-grasp-learning/HEAD/utils/visualization_utils.py --------------------------------------------------------------------------------