论坛算法工具链fcos3d模型训练的一个小问题

fcos3d模型训练的一个小问题

九zoujiu.com.cn2026-07-08
65
2
  1. 和芯片无关

  2. 和板卡无关

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

  4. fcos3d模型的训练

  5. nuscenes数据集训练fcos3d模型

开发包目录里面的两个文件,/horizon_j6_open_explorer_v3.8.1-py310_20260326/samples/ai_toolchain/horizon_model_train_sample/scripts/configs/detection/fcos3d

img_scale是错误的,正确的应该是img_scale=(512, 896),height在前,width在后,也就是(H, W),而不是wh

查看/usr/local/lib/python3.10/dist-packages/hat/data/transforms/detection.py,里面的类Resize(),函数_resize_img也是h在前w在后
算法工具链
技术深度解析征程6
评论2
0/600
  • XZH
    Lv.2
    您好,该问题属于 OpenExplorer v3.8.1 中 FCOS3D 训练样例配置的尺寸顺序不一致。Resize3D 继承自 Resize(位于 /usr/local/lib/python3.10/dist-packages/hat/data/transforms/detection.py),未重写图像 resize 的尺寸解析逻辑Resize._resize_img() 中按以下方式读取 scale:h = data["scale"][0] w = data["scale"][1]因此 Resize3D 的 img_scale 应按 (H, W) 配置。当前样例中dict(type="Resize3D", img_scale=(896, 512), keep_ratio=True) dict(type="Pad", size=(512, 896))前者表示 resize 到 (H, W) = (896, 512),而后者目标为 (H, W) = (512, 896),两者尺寸顺序不一致。建议将您提到的两个配置文件中所有 Resize3D 的 img_scale 统一修改为:img_scale=(512, 896)
    2026-07-13
    0
    0
  • HuangHui
    Lv.5

    客户您好,由于较长时间未收到您的进一步反馈,我们暂时关闭本帖。如问题仍未解决,欢迎重新发帖咨询,感谢您的理解与配合!

    2026-08-05
    0
    0