OE Agent“天工”新版本发布,请前往论坛顶部查看,欢迎试用并反馈~
查看OE-Skills
专栏算法工具链fcos3d加载训练数据不对帖子(2)附多种解决方案

fcos3d加载训练数据不对帖子(2)附多种解决方案

九zoujiu.com.cn2026-07-18
15
0
  1. 和芯片无关

  2. 和板卡无关

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

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

  5. nuscenes数据集训练fcos3d模型,输入给模型的数据存在问题,导致fcos3d模型inference出来的predict的结果是部分错误的,导致predict结果部分不可用

额外添加了两种解决方案,只需要旋转中心点到自车坐标系,然后保存最开始的yaw角就可以了的

训练加载数据的解决方案,/usr/local/lib/python3.10/dist-packages/hat/data/datasets/nuscenes_dataset.py Box类的

yaw角没有任何修改,这边收集和统计了训练数据的变换前后的yaw角,发现差距不大,差值在1左右的(弧度制)

复现和解决方案的,见附件

附件:
算法工具链
技术深度解析征程6
评论0
0/600