专栏算法工具链在J6的docker里面可以将deform_conv2d算子导出为onnx吗

在J6的docker里面可以将deform_conv2d算子导出为onnx吗

已解决
cbeibieq2025-09-26
104
19

想在j6docker中利用deform_conv2d算子构建一个简单的模型,并导出为onnx模型,导出时报错onnx export failed on an operator with unrecognized namespace torchvision::deform_conv2d,怎么解决

算法工具链
征程6
评论2
0/1000
  • Vincent
    Lv.4

    你好,这个算子地平线是支持的,下面我给您一个导出onnx的代码示例,你可以参考一下

    2025-09-26
    0
    1
    • cbeibieq回复Vincent:
      zuoyi_org.svg感谢
      2025-09-26
      0
  • Vincent
    Lv.4
    2025-09-26
    0
    16
    • cbeibieq回复Vincent:

      你好,想问下地平线版本的deform_conv和原版相比做了哪些修改呀

      2025-09-28
      0
    • Vincent回复cbeibieq:

      你好,你可以先用一个相同的输入进行原版和地平线版本的进行一个测试,看输出是否完全相同,后续我可以帮你看下这两个算子内部具体有哪些不同

      2025-09-28
      0
    • cbeibieq回复Vincent:

      测试了下相同的输入,最后输出是一致的,地平线支持 deform_conv 这个算子部署吗

      2025-09-28
      0
    • Vincent回复cbeibieq:

      支持的

      2025-09-28
      0
    • cbeibieq回复Vincent:

      可以帮忙看下这两个算子内部具体有哪些不同吗

      2025-09-28
      0
    • Vincent回复cbeibieq:

      可以的,稍后答复您

      2025-09-28
      0
    • Vincent回复cbeibieq:

      你好,刚去沟通了一下,这个是没办法去查看地平线的deformer_conv算子具体实现过程的,不过只要保证输入与输出都相同,功能应该也是都相同的。实现过程也应类似。

      2025-09-28
      0
    • cbeibieq回复Vincent:

      我尝试量化部署的时候报错custom:DeformConv2d(-1) is not a registered function/op,这个怎么解决

      2025-09-28
      0
    • Vincent回复cbeibieq:

      ptq吗还是qat

      2025-09-28
      0
    • cbeibieq回复Vincent:

      ptq

      2025-09-28
      0
    • cbeibieq回复Vincent:

      deform_conv2d和onnx里面的deform_conv是一个算子吗

      2025-09-29
      0
    • Vincent回复cbeibieq:

      你好,你用这个代码导出吧。

      2025-09-29
      0
    • Vincent回复cbeibieq:
      2025-09-29
      0
    • Vincent回复cbeibieq:

      下面发重复了有bug,你看一个就好

      2025-09-29
      0
    • Vincent回复cbeibieq:

      算子内部具体实现应该是看不了,功能都一样

      2025-09-29
      0
    • cbeibieq回复Vincent:

      好的

      2025-09-30
      0