From 209c38b95d1a8606511727e4c685b2eebcdd52a3 Mon Sep 17 00:00:00 2001 From: Benedikt Mersch Date: Mon, 14 Apr 2025 10:33:33 +0200 Subject: [PATCH] Fix --- src/mapmos/datasets/kitti.py | 3 +-- src/mapmos/odometry.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/mapmos/datasets/kitti.py b/src/mapmos/datasets/kitti.py index 9dfcaa9..9892ffe 100644 --- a/src/mapmos/datasets/kitti.py +++ b/src/mapmos/datasets/kitti.py @@ -52,8 +52,7 @@ def __init__(self, data_dir, sequence: str, *_, **__): self.label_files = sorted(glob.glob(self.label_dir + "*.label")) def __getitem__(self, idx): - points = self.scans(idx) - timestamps = np.zeros(len(points)) + points, timestamps = self.scans(idx) labels = ( self.read_labels(self.label_files[idx]) if self.label_files diff --git a/src/mapmos/odometry.py b/src/mapmos/odometry.py index 1fb2cf4..3b9e200 100644 --- a/src/mapmos/odometry.py +++ b/src/mapmos/odometry.py @@ -109,7 +109,7 @@ def deskew(self, points, timestamps, relative_motion): np.asarray( mapmos_pybind._deskew( frame=mapmos_pybind._Vector3dVector(points), - timestamps=timestamps, + timestamps=timestamps.ravel(), relative_motion=relative_motion, ) )