├── .gitignore ├── LICENSE ├── README.md ├── config ├── resnet_delta_force_mod_single_view_force_Linear_encode.yaml ├── resnet_delta_force_mod_single_view_force_MLP_encode.yaml ├── resnet_delta_force_mod_single_view_no_encode.yaml ├── resnet_delta_no_force_single_view.yaml └── resnet_delta_with_force_single_view_force_Linear_crossattn_hybrid_crop.yaml ├── data_util.py ├── env_util.py ├── imgs └── overview_system.png ├── inference.py ├── inference_real.py ├── kuka_execute.py ├── module_attr_mixin.py ├── network.py ├── real_robot_network copy.py ├── real_robot_network.py ├── requirements.txt ├── robot_data_collection_joint_state.py ├── rotation_transformer.py ├── rotation_utils.py ├── test_rotation.py ├── train.py ├── train_real.py ├── train_utils.py └── transformer_obs_encoder.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/README.md -------------------------------------------------------------------------------- /config/resnet_delta_force_mod_single_view_force_Linear_encode.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/config/resnet_delta_force_mod_single_view_force_Linear_encode.yaml -------------------------------------------------------------------------------- /config/resnet_delta_force_mod_single_view_force_MLP_encode.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/config/resnet_delta_force_mod_single_view_force_MLP_encode.yaml -------------------------------------------------------------------------------- /config/resnet_delta_force_mod_single_view_no_encode.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/config/resnet_delta_force_mod_single_view_no_encode.yaml -------------------------------------------------------------------------------- /config/resnet_delta_no_force_single_view.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/config/resnet_delta_no_force_single_view.yaml -------------------------------------------------------------------------------- /config/resnet_delta_with_force_single_view_force_Linear_crossattn_hybrid_crop.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/config/resnet_delta_with_force_single_view_force_Linear_crossattn_hybrid_crop.yaml -------------------------------------------------------------------------------- /data_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/data_util.py -------------------------------------------------------------------------------- /env_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/env_util.py -------------------------------------------------------------------------------- /imgs/overview_system.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/imgs/overview_system.png -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/inference.py -------------------------------------------------------------------------------- /inference_real.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/inference_real.py -------------------------------------------------------------------------------- /kuka_execute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/kuka_execute.py -------------------------------------------------------------------------------- /module_attr_mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/module_attr_mixin.py -------------------------------------------------------------------------------- /network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/network.py -------------------------------------------------------------------------------- /real_robot_network copy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/real_robot_network copy.py -------------------------------------------------------------------------------- /real_robot_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/real_robot_network.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/requirements.txt -------------------------------------------------------------------------------- /robot_data_collection_joint_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/robot_data_collection_joint_state.py -------------------------------------------------------------------------------- /rotation_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/rotation_transformer.py -------------------------------------------------------------------------------- /rotation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/rotation_utils.py -------------------------------------------------------------------------------- /test_rotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/test_rotation.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/train.py -------------------------------------------------------------------------------- /train_real.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/train_real.py -------------------------------------------------------------------------------- /train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/train_utils.py -------------------------------------------------------------------------------- /transformer_obs_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/JeonHoKang/Diffusion_Policy_with_Visual_Force_CrossAttn/HEAD/transformer_obs_encoder.py --------------------------------------------------------------------------------