kitti数据集训练过程的dataloader解析

illusory028

已于 2024-12-18 11:20:50 修改

阅读量978

点赞数 18

分类专栏： kitti数据集 3d检测文章标签： pytorch 人工智能 python

于 2024-09-06 11:59:03 首次发布

本文链接：https://blog.csdn.net/weixin_45994083/article/details/141954428

版权

参考的代码为：GitHub - ruhyadi/YOLO3D: YOLO 3D Object Detection for Autonomous Driving Vehicle

参考的数据集格式介绍为：KITTI数据集下载及解析_kitti标注 dontcare-CSDN博客

script.Dataset.Dataset.format_label函数的194行开始，反序列化了标注数据：

label = {
    'Class': Class,
    'Box_2D': Box_2D,
    'Dimensions': Dimension,
    'Alpha': Alpha,
    'Orientation': Orientation,
    'Confidence': Confidence
}

return label

参考txt的label文件如下：

最低0.47元/天解锁文章