├── .gitignore ├── README.md ├── datasets ├── __init__.py ├── baseline_force_dataset.py ├── keypoint_and_trajectory_dataset.py └── tweak_initial_state_dataset.py ├── environments ├── __init__.py ├── base_env.py ├── bullet_client.py ├── env_wrapper_multiple_object.py └── physics_env.py ├── figs └── teaser_force.jpg ├── main.py ├── models ├── __init__.py ├── base_model.py ├── baseline_regress_force.py ├── gt_cp_predict_init_pose_and_force.py ├── image_and_cp_input_model_keypoint_predict.py ├── image_input_predict_cp_separate_tower.py ├── no_model_gt_calculator.py └── only_predict_cp_model.py ├── requirements.txt ├── scripts ├── test_w_weights.sh └── train_scripts.sh ├── solvers ├── metrics.py ├── save_gt_force.py ├── test.py └── train.py └── utils ├── arg_parser.py ├── constants.py ├── data_loading_utils.py ├── environment_util.py ├── logging_util.py ├── loss_util.py ├── net_util.py ├── obj_util.py ├── object_list.txt ├── projection_utils.py ├── quaternion_util.py ├── transformations.py └── visualization_util.py /.gitignore: -------------------------------------------------------------------------------- 1 | .vscode/ 2 | data/ 3 | __pycache__ 4 | .DS_Store 5 | FPHA 6 | ycb 7 | .idea 8 | DatasetForce -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/README.md -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/baseline_force_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/datasets/baseline_force_dataset.py -------------------------------------------------------------------------------- /datasets/keypoint_and_trajectory_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/datasets/keypoint_and_trajectory_dataset.py -------------------------------------------------------------------------------- /datasets/tweak_initial_state_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/datasets/tweak_initial_state_dataset.py -------------------------------------------------------------------------------- /environments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/environments/__init__.py -------------------------------------------------------------------------------- /environments/base_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/environments/base_env.py -------------------------------------------------------------------------------- /environments/bullet_client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/environments/bullet_client.py -------------------------------------------------------------------------------- /environments/env_wrapper_multiple_object.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/environments/env_wrapper_multiple_object.py -------------------------------------------------------------------------------- /environments/physics_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/environments/physics_env.py -------------------------------------------------------------------------------- /figs/teaser_force.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/figs/teaser_force.jpg -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/models/base_model.py -------------------------------------------------------------------------------- /models/baseline_regress_force.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/models/baseline_regress_force.py -------------------------------------------------------------------------------- /models/gt_cp_predict_init_pose_and_force.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/models/gt_cp_predict_init_pose_and_force.py -------------------------------------------------------------------------------- /models/image_and_cp_input_model_keypoint_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/models/image_and_cp_input_model_keypoint_predict.py -------------------------------------------------------------------------------- /models/image_input_predict_cp_separate_tower.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/models/image_input_predict_cp_separate_tower.py -------------------------------------------------------------------------------- /models/no_model_gt_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/models/no_model_gt_calculator.py -------------------------------------------------------------------------------- /models/only_predict_cp_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/models/only_predict_cp_model.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/test_w_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/scripts/test_w_weights.sh -------------------------------------------------------------------------------- /scripts/train_scripts.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/scripts/train_scripts.sh -------------------------------------------------------------------------------- /solvers/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/solvers/metrics.py -------------------------------------------------------------------------------- /solvers/save_gt_force.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/solvers/save_gt_force.py -------------------------------------------------------------------------------- /solvers/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/solvers/test.py -------------------------------------------------------------------------------- /solvers/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/solvers/train.py -------------------------------------------------------------------------------- /utils/arg_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/utils/arg_parser.py -------------------------------------------------------------------------------- /utils/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/utils/constants.py -------------------------------------------------------------------------------- /utils/data_loading_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/utils/data_loading_utils.py -------------------------------------------------------------------------------- /utils/environment_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/utils/environment_util.py -------------------------------------------------------------------------------- /utils/logging_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/utils/logging_util.py -------------------------------------------------------------------------------- /utils/loss_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/utils/loss_util.py -------------------------------------------------------------------------------- /utils/net_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/utils/net_util.py -------------------------------------------------------------------------------- /utils/obj_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/utils/obj_util.py -------------------------------------------------------------------------------- /utils/object_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/utils/object_list.txt -------------------------------------------------------------------------------- /utils/projection_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/utils/projection_utils.py -------------------------------------------------------------------------------- /utils/quaternion_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/utils/quaternion_util.py -------------------------------------------------------------------------------- /utils/transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/utils/transformations.py -------------------------------------------------------------------------------- /utils/visualization_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ehsanik/touchTorch/HEAD/utils/visualization_util.py --------------------------------------------------------------------------------