Skip to content

Commit 2fe256e

Browse files
authored
fix kitti data sequence parsing (#80)
1 parent 0fe822d commit 2fe256e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/map_closures/datasets/kitti.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,15 +28,15 @@
2828

2929
class KITTIOdometryDataset:
3030
def __init__(self, data_dir, sequence: str, *_, **__):
31-
self.sequence_id = sequence
31+
self.sequence_id = str(sequence).zfill(2)
3232
self.sequence_dir = os.path.join(data_dir, "sequences", self.sequence_id)
3333
self.velodyne_dir = os.path.join(self.sequence_dir, "velodyne/")
3434

3535
self.scan_files = sorted(glob.glob(self.velodyne_dir + "*.bin"))
3636
self.calibration = self.read_calib_file(os.path.join(self.sequence_dir, "calib.txt"))
3737

3838
# Load GT Poses (if available)
39-
if sequence < 11:
39+
if int(sequence) < 11:
4040
self.poses_fn = os.path.join(data_dir, f"poses/{self.sequence_id}.txt")
4141
self.gt_poses = self.load_poses(self.poses_fn)
4242

0 commit comments

Comments
 (0)