专栏算法工具链fcos3d 模型导出浮点模型，配置文件如何修改

fcos3d 模型导出浮点模型，配置文件如何修改

加利福尼亚不下雨15小时前
17
1

OE包：horizon_j6_open_explorer_v3.0.17-py310_20240705

文件路径：/open_explorer/samples/ai_toolchain/horizon_model_train_sample/scripts/configs/detection/fcos3d/fcos3d_efficientnetb0_nuscenes.py

内容：

上面是导出qat模型的配置，导出浮点模型，如何修改？

算法工具链
技术深度解析征程6
评论1
0/1000
  • Huanghui
    Lv.5
    你好，onnx导出逻辑主要是使用脚本scripts/tools/export_onnx.py以及模型配置文件中的onnx_cfg完成模型导出，从scripts/tools/export_onnx.py的代码逻辑可以看出导出模型由onnc_cfg中的stage项决定：

    你可以尝试使用如下的配置导出浮点onnx：

    需要注意：模型中不应含有地平线专有算子，否则即使导出也无法通过onneruntime等使用
    15小时前
    0
    0