专栏算法工具链编译生成hbm模型，PointPillarsProPrecess算子输入如何改为tensor类型？

编译生成hbm模型，PointPillarsProPrecess算子输入如何改为tensor类型？

chengang2023-08-29
32
2
1.芯片型号：J5
2.天工开物开发包OpenExplorer版本： J5_OE_1.1.57
3.问题定位：编译生成hbm模型
4.问题具体描述：

你好，我们开发了一个雷达和图像融合的模型，deploy_model包含PointPillarsProPrecess算子。

参看pointpillars算法，deploy_inputs中points要求类型为List[Tensor]，但是融合模型图像和采样点数据类型为tensor，编译时会提示: Dictionary inputs to traced function must have consistent type, Found Tensor and List[Tensor]。

这里有什么办法修改PointPillarsProPrecess算子输入类型为tensor吗？

算法工具链
评论1
0/1000
  • 颜值即正义
    Lv.2
    您好，建议把tensor放在list中，在底层的_point_pillars_preprocess中会遍历list做voxelization，或者修改plugin的代码/usr/local/lib/python3.8/dist-packages/horizon_plugin_pytorch/nn/quantized/functional_impl.py的_point_pillars_preprocess
    2023-08-29
    0
    1
    • chengang回复颜值即正义:
      问题解决了，在deploy_inputs中将所有输入改为List[Tensor]。plugin中的_point_pillars_preprocess函数执行前会执行check_argument_types函数，要求PointPillarsProPrecess输入为List[Tensor]。
      2023-08-29
      0