tartanimu-urad-unified

URAD — TartanIMU Challenge (IROS 2026) single unified model

One network, one set of weights, all four platforms (car, quadruped, drone, handheld). Raw 200 Hz IMU in, one body-frame velocity per one-second window out.

Run

pip install -r requirements.txt            # or requirements-cpu.txt
python inference.py --data <competition data root> --checkpoint model.pt \
       --reference submission.csv --out out --device cuda

<competition data root> must contain index/test_windows.csv, sample_submission.csv and test/*.npz. No network access is needed. The script writes out/submission.csv and a JSON audit with the maximum absolute difference to the reference (our CPU re-run: 3.97e-6 m/s, see runtime_cpu.json).

What the model is

A small residual 1-D CNN, the organizers' pretrained TartanIMU ResNet-1D trunk used as a second feature branch, and window / gyro-rotation / whole-trajectory motion descriptors feed a two-layer bidirectional GRU over 64 windows; a learned softmax gate mixes four jointly trained velocity heads inside the same network.

Disclosures

• Checkpoint averaging: model.pt is the parameter-wise mean of five fine-tuning runs from one common checkpoint.
• Inference-time processing: overlapping 64-window contexts of this one checkpoint are averaged at stride 8, and every output vector is multiplied by 1.02. Both settings are stored in the checkpoint config (infer_stride, output_scale) and applied by inference.py. No input-transforming TTA, smoothing, filtering or clipping.
• Platform handling: internal learned gate only. No platform label, trajectory identity, pose or ground truth enters the forward pass.
• Training data: the challenge train and val splits. No external data. Test IMU was never used for fitting.
• Pretrained weights: the organizers' Tartan-IMU/TartanIMU checkpoints/unified.pt (revision 556a9e89c85d35ad83b7d05c852194165a6601cd), fine-tuned and embedded in model.pt; see EXTERNAL_PRETRAINING.json.
• Inference is offline and non-causal (whole-trajectory descriptors), and each trajectory is processed on its own.

Licence and attribution

Model and code definitions derive from TartanIMU (copyright 2026 Shibo Zhao, Apache-2.0; licence text in TartanIMU-LICENSE). The embedded weights derive from the organizers' released checkpoint, whose model card restricts use to research / non-commercial purposes; this repository carries the same restriction. See NOTICE.