专栏算法工具链模型转换

模型转换

已解决
onlhy2024-07-24
43
4

模型转化

网络中间夹杂了nms非极大值抑制这种能转换吗，具体就是图片输入经过网络特征提取，得到的特征图经过一个nms非极大值抑制，之后再送进下一阶段的网络，然后得到输出，这种情况是把网络拆分成两个onnx转换，前面用图片输入，后面用featuremap做为输入吗，还是说可以把中间的nms可以变成类似算子一样的操作，之间输出单个onnx呢
算法工具链
征程5
评论3
0/1000
  • GOYU\XU
    Lv.2

    通过yolo系列转换过程中nms操作这个地方会报错，中间有nms的操作，按你说的分为两个onnx的思路我认为是可以的

    2024-07-24
    0
    1
    • onlhy回复GOYU\XU:

      地平线有这样的例子可以参考吗

      2024-07-24
      0
  • GOYU\XU
    Lv.2

    目前没有nms这种分段的onnx示例

    2024-07-24
    0
    0
  • GOYU\XU
    Lv.2

    客户您好，鉴于长时间未收到你的答复，我们暂将该问题做关闭处理，如您仍有相同问题，请重新开帖子进行提问，祝您工作愉快！

    2024-08-06
    0
    0