结论
目前地平线工具链不支持稀疏卷积(spconv)算子,官方建议进行算子替换,将稀疏卷积转化为标准3D卷积或其他支持的算子进行部署。
解决方案
1. 现有资源参考
社区已有成熟的点云算法参考实现,可直接参考:
- PointPillars参考算法 - 包含点云预处理和标准卷积实现
- CenterPoint参考算法 - 3D检测完整方案
2. 算子替换策略
# 稀疏卷积 → 标准卷积的转换思路
# 方案1: 将稀疏体素网格转换为稠密网格
# 方案2: 使用PointPillarsScatter等支持的算子替代
# 方案3: 修改网络结构,使用标准3D卷积层
3. 验证算子支持
查看官方算子支持列表:
- S100/S100P ONNX算子支持:S100/S100P ONNX 算子支持列表 - 地瓜算法工具链官方用户手册
- S100/S100P Torch算子支持:S100/S100P Torch 算子支持列表 - 地瓜算法工具链官方用户手册
4. 建议步骤
- 导出ONNX模型,使用
hb_compile工具检查算子支持情况 - 定位不支持的算子,查看报错信息
- 修改网络结构,替换为支持的算子组合
- 参考PointPillars实现,采用已验证的架构
社区交流
话题中已有开发者在研究类似问题,可以:
- 在该帖子下继续讨论技术细节
- 参考PointPillars/CenterPoint的开源实现
- 联系官方技术支持获取定制方案(如为商业项目)
需要我帮你查看具体的算子替换方案或PointPillars的实现细节吗?