NuScenes-devkit中KITTI数据集格式转换问题解析

背景介绍

在自动驾驶领域，KITTI和NuScenes是两个广泛使用的数据集。NuScenes-devkit项目提供了一个名为export_kitti.py的脚本，旨在帮助用户在两种数据集格式之间进行转换。然而，许多开发者在尝试将KITTI数据集转换为NuScenes格式时遇到了困难。

常见问题分析

1. 路径配置问题

开发者经常遇到的首要问题是路径配置错误。当使用export_kitti.py脚本时，需要正确设置以下参数：

• nusc_kitti_dir：指向包含KITTI数据的根目录
• split：指定要处理的数据分割（如'train'或'test'）
• nusc_version：指定NuScenes数据版本

常见错误包括：

• 文件夹命名不符合预期
• 路径结构不正确
• 版本号不匹配

2. 脚本功能误解

许多开发者误以为export_kitti.py可以直接将原生KITTI数据转换为NuScenes格式。实际上，该脚本设计初衷是将NuScenes数据以KITTI格式导出，或者处理已经以KITTI格式存储的NuScenes数据。

3. 数据版本兼容性

当使用不同版本的NuScenes数据集时（如'v1.0-mini'或'v1.0-trainval'），需要确保：

• 数据目录结构正确
• 配置文件版本匹配
• 分割名称一致（如'mini_train'对应v1.0-mini）

解决方案

1. 目录结构调整

确保KITTI数据目录结构如下：

KITTI
├── maps
├── render
├── samples
├── sweeps
├── test
├── train
└── v1.0-mini

2. 参数正确配置

根据使用的数据集版本，正确设置参数：

class KittiConverter:
    def __init__(self,
                 nusc_kitti_dir: str = '/KITTI',
                 cam_name: str = 'CAM_FRONT',
                 lidar_name: str = 'LIDAR_TOP',
                 image_count: int = 10,
                 nusc_version: str = 'v1.0-trainval',  # 或'v1.0-mini'
                 split: str = 'train'):  # 或'mini_train'

3. 脚本修改建议

如果需要将原生KITTI数据转换为NuScenes格式，可以修改脚本中的样本token获取逻辑：

# 原始代码使用NuScenes样本token
sample_tokens = [s['token'] for s in self.nusc.sample]

# 修改为直接使用KITTI tokens
sample_tokens = kitti.tokens[:self.image_count]

技术要点

1. 数据格式差异：理解KITTI和NuScenes在数据组织、标注格式和坐标系方面的差异是成功转换的关键。

2. 依赖关系：即使只是转换KITTI数据，NuScenes-devkit提供的工具类和函数也是必要的，因为它包含了处理点云、图像和标注的标准方法。

3. 扩展性考虑：对于大规模数据转换，需要考虑内存管理和批处理优化。

最佳实践

1. 始终验证输入数据的完整性和路径正确性
2. 对于自定义转换需求，考虑继承或修改KittiDB类
3. 转换完成后，使用NuScenes可视化工具验证结果
4. 对于大批量数据，实现进度跟踪和错误恢复机制

通过理解这些技术细节和解决方案，开发者可以更有效地在KITTI和NuScenes格式之间进行转换，为自动驾驶算法开发提供更灵活的数据支持。