OpenPCDet模型导出ONNX

前面介绍了采用OpenPCDet架构和自定义数据集训练了pointRCNN模型,也已经跑通了测试流程,但在实际应用过程中通常需要把模型导出成ONNX格式,然后借助其他框架用C++进行推理。这里记录一下pointRCNN导出流程以及常见的一些错误。其大体分为以下几步: 弄清楚数据流 确定网络架构 改写
posted @ 2024-06-02 15:44  半夜打老虎  阅读(19)  评论(0编辑  收藏  举报