提供两支轻量级Python脚本,替代原MATLAB版SUN RGB-D数据集预处理流程,专为mmdetection3d框架优化。extract_split.py负责按官方划分生成train/val/test子集的图像与深度图路径索引;extract_data_v2.py解析原始.mat标注文件,提取2D边界框、3D朝向、类别ID、实例分割掩码等结构化信息,并统一转为COCO-style JSON格式,直接兼容mmdetection3d的数据加载器。所有脚本不依赖MATLAB环境,仅需Python 3.7+、scipy、numpy、Pillow等基础库,支持Linux/macOS/Windows平台运行。目录中scripts文件夹预留扩展位,可用于后续添加数据增强、可视化或格式转换工具。使用前建议确认SUN RGB-D原始数据已按标准目录结构解压,脚本内含清晰注释与路径配置说明,可快速对接自定义训练流程。
1