OE Agent“天工”新版本发布,请前往论坛顶部查看,欢迎试用并反馈~
查看OE-Skills
专栏算法工具链加载lmdb训练数据过滤3D框的codes多过滤了一部分3D框

加载lmdb训练数据过滤3D框的codes多过滤了一部分3D框

九zoujiu.com.cn2026-07-04
10
1
  1. 和芯片无关

  2. 和板卡无关

  3. 天工开物开发包 OpenExplorer 版本:openexplorer/ai_toolchain_ubuntu_22_j6_gpu:v3.8.1,容器docker内开发

  4. fcos3d模型的训练 + 训练数据集3D框可视化 + predict可视化

  5. nuscenes数据集训练fcos3d模型,训练bev模型的

训练时加载数据,过滤框的机制多过滤了一些框,具体见附件的,附件的图片中,框多的左侧是自己的,框少的右侧是地平线的

相关的codes是目录/usr/local/lib/python3.10/dist-packages/hat/data/datasets中的文件nuscenes_dataset.py,类class NuscenesSample(object):,函数def get_mono_by_name(self, name):

相应的其他几个过滤函数的部分应该也存在或多或少的问题,也就是多过滤了一些3D框

附件:
算法工具链
技术深度解析征程6
评论1
0/600
  • YCJ
    Lv.4

    从截图看,部分近处、画面边缘、严重截断的车辆没有进入右侧结果,可能符合单目三维检测训练数据的有效性筛选逻辑。这类目标通常存在可见区域不完整、投影越界较多、距离相机过近、投影结果不稳定等情况,不一定适合作为当前相机视角下的有效训练标注。

    但图中清晰可见的行人或完整目标如果也没有进入右侧结果,则建议重点排查类别映射和后续数据处理流程。请确认客户数据中的类别名称是否和训练配置一致,例如行人、其他障碍物、三轮车、两轮车等类别是否都已加入类别映射表和训练类别表。

    仅凭截图无法判断每个框未进入结果的具体原因。建议在单目数据读取函数中增加本地调试日志,选几张有问题的样本,输出未进入结果框的原始类别、映射类别编号、中心深度、二维投影框坐标、投影框宽高和具体原因。

    如果主要原因是类别未映射,需要补齐类别映射;如果是深度或投影异常,需要检查相机内外参、相机名称、图像去畸变和三维框坐标系;如果是图像外、近处截断或投影大面积越界,则更可能属于正常筛选。

    11小时前
    0
    0