├── .github └── workflows │ ├── formatting.yaml │ └── pytest.yaml ├── .gitignore ├── .header.txt ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── assets ├── docs │ ├── feature_prediction.png │ ├── gso_vs_svd1.gif │ ├── gso_vs_svd2.gif │ ├── gso_vs_svd3.gif │ ├── overview_bottom.png │ ├── overview_top.png │ ├── rotation_estimation.png │ ├── torus.png │ └── torus_v5.pdf └── prettyplots.mplstyle ├── hitchhiking_rotations ├── __init__.py ├── cfgs │ ├── __init__.py │ ├── cfg_cube_image_to_pose.py │ ├── cfg_pcd_to_pose.py │ ├── cfg_pose_to_cube_image.py │ └── cfg_pose_to_fourier.py ├── datasets │ ├── .cube_data.swp │ ├── __init__.py │ ├── cube_data_generator.py │ ├── cube_dataset.py │ ├── fourier_dataset.py │ └── pointcloud_dataset.py ├── models │ ├── __init__.py │ └── models.py └── utils │ ├── __init__.py │ ├── colors.py │ ├── conversions.py │ ├── euler_helper.py │ ├── helper.py │ ├── loading.py │ ├── logger.py │ ├── metrics.py │ ├── notation.py │ ├── trainer.py │ └── training_helper.py ├── results └── dense_fusion │ └── dense_fusion_experiment.csv ├── scripts ├── run_all.py └── train.py ├── setup.py ├── tests ├── test_roma_verify_rotations.py └── test_svd_timing.py └── visu ├── figure_exp2a.py ├── figure_exp2b.py ├── figure_fourier.py ├── figure_mocap.py ├── figure_pointcloud.py ├── figure_posenet.py ├── gso_vs_svd.py ├── gso_vs_svd_animation.py ├── lipschitz_constants.py ├── loss_gradients.py ├── tables_21.py ├── time_network.py └── time_only_svd.py /.github/workflows/formatting.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/.github/workflows/formatting.yaml -------------------------------------------------------------------------------- /.github/workflows/pytest.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/.github/workflows/pytest.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/.gitignore -------------------------------------------------------------------------------- /.header.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/.header.txt -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/README.md -------------------------------------------------------------------------------- /assets/docs/feature_prediction.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/assets/docs/feature_prediction.png -------------------------------------------------------------------------------- /assets/docs/gso_vs_svd1.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/assets/docs/gso_vs_svd1.gif -------------------------------------------------------------------------------- /assets/docs/gso_vs_svd2.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/assets/docs/gso_vs_svd2.gif -------------------------------------------------------------------------------- /assets/docs/gso_vs_svd3.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/assets/docs/gso_vs_svd3.gif -------------------------------------------------------------------------------- /assets/docs/overview_bottom.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/assets/docs/overview_bottom.png -------------------------------------------------------------------------------- /assets/docs/overview_top.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/assets/docs/overview_top.png -------------------------------------------------------------------------------- /assets/docs/rotation_estimation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/assets/docs/rotation_estimation.png -------------------------------------------------------------------------------- /assets/docs/torus.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/assets/docs/torus.png -------------------------------------------------------------------------------- /assets/docs/torus_v5.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/assets/docs/torus_v5.pdf -------------------------------------------------------------------------------- /assets/prettyplots.mplstyle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/assets/prettyplots.mplstyle -------------------------------------------------------------------------------- /hitchhiking_rotations/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/__init__.py -------------------------------------------------------------------------------- /hitchhiking_rotations/cfgs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/cfgs/__init__.py -------------------------------------------------------------------------------- /hitchhiking_rotations/cfgs/cfg_cube_image_to_pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/cfgs/cfg_cube_image_to_pose.py -------------------------------------------------------------------------------- /hitchhiking_rotations/cfgs/cfg_pcd_to_pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/cfgs/cfg_pcd_to_pose.py -------------------------------------------------------------------------------- /hitchhiking_rotations/cfgs/cfg_pose_to_cube_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/cfgs/cfg_pose_to_cube_image.py -------------------------------------------------------------------------------- /hitchhiking_rotations/cfgs/cfg_pose_to_fourier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/cfgs/cfg_pose_to_fourier.py -------------------------------------------------------------------------------- /hitchhiking_rotations/datasets/.cube_data.swp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/datasets/.cube_data.swp -------------------------------------------------------------------------------- /hitchhiking_rotations/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/datasets/__init__.py -------------------------------------------------------------------------------- /hitchhiking_rotations/datasets/cube_data_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/datasets/cube_data_generator.py -------------------------------------------------------------------------------- /hitchhiking_rotations/datasets/cube_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/datasets/cube_dataset.py -------------------------------------------------------------------------------- /hitchhiking_rotations/datasets/fourier_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/datasets/fourier_dataset.py -------------------------------------------------------------------------------- /hitchhiking_rotations/datasets/pointcloud_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/datasets/pointcloud_dataset.py -------------------------------------------------------------------------------- /hitchhiking_rotations/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/models/__init__.py -------------------------------------------------------------------------------- /hitchhiking_rotations/models/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/models/models.py -------------------------------------------------------------------------------- /hitchhiking_rotations/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/utils/__init__.py -------------------------------------------------------------------------------- /hitchhiking_rotations/utils/colors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/utils/colors.py -------------------------------------------------------------------------------- /hitchhiking_rotations/utils/conversions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/utils/conversions.py -------------------------------------------------------------------------------- /hitchhiking_rotations/utils/euler_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/utils/euler_helper.py -------------------------------------------------------------------------------- /hitchhiking_rotations/utils/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/utils/helper.py -------------------------------------------------------------------------------- /hitchhiking_rotations/utils/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/utils/loading.py -------------------------------------------------------------------------------- /hitchhiking_rotations/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/utils/logger.py -------------------------------------------------------------------------------- /hitchhiking_rotations/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/utils/metrics.py -------------------------------------------------------------------------------- /hitchhiking_rotations/utils/notation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/utils/notation.py -------------------------------------------------------------------------------- /hitchhiking_rotations/utils/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/utils/trainer.py -------------------------------------------------------------------------------- /hitchhiking_rotations/utils/training_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/hitchhiking_rotations/utils/training_helper.py -------------------------------------------------------------------------------- /results/dense_fusion/dense_fusion_experiment.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/results/dense_fusion/dense_fusion_experiment.csv -------------------------------------------------------------------------------- /scripts/run_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/scripts/run_all.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/scripts/train.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_roma_verify_rotations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/tests/test_roma_verify_rotations.py -------------------------------------------------------------------------------- /tests/test_svd_timing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/tests/test_svd_timing.py -------------------------------------------------------------------------------- /visu/figure_exp2a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/visu/figure_exp2a.py -------------------------------------------------------------------------------- /visu/figure_exp2b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/visu/figure_exp2b.py -------------------------------------------------------------------------------- /visu/figure_fourier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/visu/figure_fourier.py -------------------------------------------------------------------------------- /visu/figure_mocap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/visu/figure_mocap.py -------------------------------------------------------------------------------- /visu/figure_pointcloud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/visu/figure_pointcloud.py -------------------------------------------------------------------------------- /visu/figure_posenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/visu/figure_posenet.py -------------------------------------------------------------------------------- /visu/gso_vs_svd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/visu/gso_vs_svd.py -------------------------------------------------------------------------------- /visu/gso_vs_svd_animation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/visu/gso_vs_svd_animation.py -------------------------------------------------------------------------------- /visu/lipschitz_constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/visu/lipschitz_constants.py -------------------------------------------------------------------------------- /visu/loss_gradients.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/visu/loss_gradients.py -------------------------------------------------------------------------------- /visu/tables_21.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /visu/time_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/visu/time_network.py -------------------------------------------------------------------------------- /visu/time_only_svd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/martius-lab/hitchhiking-rotations/HEAD/visu/time_only_svd.py --------------------------------------------------------------------------------