稀疏卷积/3D卷积部署问题

结论

目前地平线工具链不支持稀疏卷积(spconv)算子,官方建议进行算子替换,将稀疏卷积转化为标准3D卷积或其他支持的算子进行部署。

解决方案

1. 现有资源参考

社区已有成熟的点云算法参考实现,可直接参考:

2. 算子替换策略

# 稀疏卷积 → 标准卷积的转换思路
# 方案1: 将稀疏体素网格转换为稠密网格
# 方案2: 使用PointPillarsScatter等支持的算子替代
# 方案3: 修改网络结构,使用标准3D卷积层

3. 验证算子支持

查看官方算子支持列表:

4. 建议步骤

  1. 导出ONNX模型,使用hb_compile工具检查算子支持情况
  2. 定位不支持的算子,查看报错信息
  3. 修改网络结构,替换为支持的算子组合
  4. 参考PointPillars实现,采用已验证的架构

社区交流

话题中已有开发者在研究类似问题,可以:

  • 在该帖子下继续讨论技术细节
  • 参考PointPillars/CenterPoint的开源实现
  • 联系官方技术支持获取定制方案(如为商业项目)

需要我帮你查看具体的算子替换方案或PointPillars的实现细节吗?