This project implements a deep learning-based Visual-Inertial Odometry (VIO) system using PyTorch. It includes three model types: Vision-only, Inertial-only, and Fusion-based models, each using LSTM networks to estimate poses from simulated or real sensor data.
- VisionOnlyModel: LSTM model processing vision features.
- InertialOnlyModel: LSTM model processing IMU sequences.
- FusionModel: Combines both vision and inertial inputs through dedicated LSTMs and concatenates their outputs for pose estimation.
vio.pyโ Core model definitions for VIO using PyTorch.
- Python 3.8+
- PyTorch
- NumPy
Install dependencies using:
pip install torch numpy