├── .gitignore ├── LICENSE ├── README.md ├── conda_environment.yaml ├── datasets └── Blackbird │ ├── clover │ └── yawForward │ │ └── maxSpeed5p0 │ │ ├── test │ │ ├── data.hdf5 │ │ └── stamped_groundtruth_imu.txt │ │ ├── train │ │ ├── data.hdf5 │ │ └── stamped_groundtruth_imu.txt │ │ └── val │ │ ├── data.hdf5 │ │ └── stamped_groundtruth_imu.txt │ ├── test.txt │ ├── train.txt │ └── val.txt ├── img └── intro_github.png ├── results └── Blackbird │ └── checkpoints │ └── model_net │ ├── model_net_parameters_net_blackbird.json │ └── net_blackbird.pt └── src ├── filter └── python │ ├── plot_filter_output.py │ └── src │ ├── .gitignore │ ├── data_io.py │ ├── filter_manager.py │ ├── filter_runner.py │ ├── meas_source_network.py │ ├── net_input_utils.py │ ├── scekf.py │ └── utils │ ├── .gitignore │ ├── argparse_utils.py │ ├── dotdict.py │ ├── from_scipy.py │ ├── logging.py │ ├── math_utils.py │ ├── misc.py │ ├── plotting.py │ └── profile.py ├── learning ├── .gitignore ├── data_management │ ├── .gitignore │ ├── datasets.py │ └── prepare_datasets │ │ ├── .gitignore │ │ ├── blackbird.py │ │ └── utils.py ├── network │ ├── .gitignore │ ├── losses.py │ ├── model_factory.py │ └── model_tcn.py ├── test_model_net.py ├── train_model_net.py └── utils │ ├── argparse_utils.py │ ├── imu.py │ ├── logging.py │ ├── math.py │ ├── model.py │ ├── plot_utils.py │ ├── pose.py │ ├── torch_math.py │ └── transformations.py ├── main_filter.py └── main_learning.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea/ 2 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/README.md -------------------------------------------------------------------------------- /conda_environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/conda_environment.yaml -------------------------------------------------------------------------------- /datasets/Blackbird/clover/yawForward/maxSpeed5p0/test/data.hdf5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/datasets/Blackbird/clover/yawForward/maxSpeed5p0/test/data.hdf5 -------------------------------------------------------------------------------- /datasets/Blackbird/clover/yawForward/maxSpeed5p0/test/stamped_groundtruth_imu.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/datasets/Blackbird/clover/yawForward/maxSpeed5p0/test/stamped_groundtruth_imu.txt -------------------------------------------------------------------------------- /datasets/Blackbird/clover/yawForward/maxSpeed5p0/train/data.hdf5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/datasets/Blackbird/clover/yawForward/maxSpeed5p0/train/data.hdf5 -------------------------------------------------------------------------------- /datasets/Blackbird/clover/yawForward/maxSpeed5p0/train/stamped_groundtruth_imu.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/datasets/Blackbird/clover/yawForward/maxSpeed5p0/train/stamped_groundtruth_imu.txt -------------------------------------------------------------------------------- /datasets/Blackbird/clover/yawForward/maxSpeed5p0/val/data.hdf5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/datasets/Blackbird/clover/yawForward/maxSpeed5p0/val/data.hdf5 -------------------------------------------------------------------------------- /datasets/Blackbird/clover/yawForward/maxSpeed5p0/val/stamped_groundtruth_imu.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/datasets/Blackbird/clover/yawForward/maxSpeed5p0/val/stamped_groundtruth_imu.txt -------------------------------------------------------------------------------- /datasets/Blackbird/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/datasets/Blackbird/test.txt -------------------------------------------------------------------------------- /datasets/Blackbird/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/datasets/Blackbird/train.txt -------------------------------------------------------------------------------- /datasets/Blackbird/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/datasets/Blackbird/val.txt -------------------------------------------------------------------------------- /img/intro_github.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/img/intro_github.png -------------------------------------------------------------------------------- /results/Blackbird/checkpoints/model_net/model_net_parameters_net_blackbird.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/results/Blackbird/checkpoints/model_net/model_net_parameters_net_blackbird.json -------------------------------------------------------------------------------- /results/Blackbird/checkpoints/model_net/net_blackbird.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/results/Blackbird/checkpoints/model_net/net_blackbird.pt -------------------------------------------------------------------------------- /src/filter/python/plot_filter_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/plot_filter_output.py -------------------------------------------------------------------------------- /src/filter/python/src/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | -------------------------------------------------------------------------------- /src/filter/python/src/data_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/data_io.py -------------------------------------------------------------------------------- /src/filter/python/src/filter_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/filter_manager.py -------------------------------------------------------------------------------- /src/filter/python/src/filter_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/filter_runner.py -------------------------------------------------------------------------------- /src/filter/python/src/meas_source_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/meas_source_network.py -------------------------------------------------------------------------------- /src/filter/python/src/net_input_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/net_input_utils.py -------------------------------------------------------------------------------- /src/filter/python/src/scekf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/scekf.py -------------------------------------------------------------------------------- /src/filter/python/src/utils/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | -------------------------------------------------------------------------------- /src/filter/python/src/utils/argparse_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/utils/argparse_utils.py -------------------------------------------------------------------------------- /src/filter/python/src/utils/dotdict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/utils/dotdict.py -------------------------------------------------------------------------------- /src/filter/python/src/utils/from_scipy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/utils/from_scipy.py -------------------------------------------------------------------------------- /src/filter/python/src/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/utils/logging.py -------------------------------------------------------------------------------- /src/filter/python/src/utils/math_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/utils/math_utils.py -------------------------------------------------------------------------------- /src/filter/python/src/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/utils/misc.py -------------------------------------------------------------------------------- /src/filter/python/src/utils/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/utils/plotting.py -------------------------------------------------------------------------------- /src/filter/python/src/utils/profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/filter/python/src/utils/profile.py -------------------------------------------------------------------------------- /src/learning/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/.gitignore -------------------------------------------------------------------------------- /src/learning/data_management/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | -------------------------------------------------------------------------------- /src/learning/data_management/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/data_management/datasets.py -------------------------------------------------------------------------------- /src/learning/data_management/prepare_datasets/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | -------------------------------------------------------------------------------- /src/learning/data_management/prepare_datasets/blackbird.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/data_management/prepare_datasets/blackbird.py -------------------------------------------------------------------------------- /src/learning/data_management/prepare_datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/data_management/prepare_datasets/utils.py -------------------------------------------------------------------------------- /src/learning/network/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | -------------------------------------------------------------------------------- /src/learning/network/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/network/losses.py -------------------------------------------------------------------------------- /src/learning/network/model_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/network/model_factory.py -------------------------------------------------------------------------------- /src/learning/network/model_tcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/network/model_tcn.py -------------------------------------------------------------------------------- /src/learning/test_model_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/test_model_net.py -------------------------------------------------------------------------------- /src/learning/train_model_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/train_model_net.py -------------------------------------------------------------------------------- /src/learning/utils/argparse_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/utils/argparse_utils.py -------------------------------------------------------------------------------- /src/learning/utils/imu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/utils/imu.py -------------------------------------------------------------------------------- /src/learning/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/utils/logging.py -------------------------------------------------------------------------------- /src/learning/utils/math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/utils/math.py -------------------------------------------------------------------------------- /src/learning/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/utils/model.py -------------------------------------------------------------------------------- /src/learning/utils/plot_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/utils/plot_utils.py -------------------------------------------------------------------------------- /src/learning/utils/pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/utils/pose.py -------------------------------------------------------------------------------- /src/learning/utils/torch_math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/utils/torch_math.py -------------------------------------------------------------------------------- /src/learning/utils/transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/learning/utils/transformations.py -------------------------------------------------------------------------------- /src/main_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/main_filter.py -------------------------------------------------------------------------------- /src/main_learning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/uzh-rpg/learned_inertial_model_odometry/HEAD/src/main_learning.py --------------------------------------------------------------------------------