pth转onnx报错

一、问题现象（附报错日志上下文）：
https://gitee.com/ascend/ModelZoo-PyTorch/tree/master/ACL_PyTorch/contrib/cv/tracking/FairMOT
从pth转onnx报错
![输入图片说明](https://foruda.gitee.com/images/1693987341883068143/c1e65d4e_12163691.png "屏幕截图")
![输入图片说明](https://foruda.gitee.com/images/1693987360895794473/14eba1cd_12163691.png "屏幕截图")
![输入图片说明](https://foruda.gitee.com/images/1693987228921577627/e3a92a10_12163691.png "屏幕截图")
二、软件版本:
-- CANN 版本 (e.g., CANN 3.0.x，5.x.x):  6.3RC1
--Tensorflow/Pytorch/MindSpore 版本: torch 1.5
--Python 版本 (e.g., Python 3.7.5):3.7.5
-- MindStudio版本 (e.g., MindStudio 2.0.0 (beta3)):无
--操作系统版本 (e.g., Ubuntu 18.04):ubuntu 18.04

三、测试步骤：
python3 /root/ModelZoo-PyTorch-master/ACL_PyTorch/contrib/cv/tracking/FairMOT/fairmot_pth2onnx.py --input_file=/root/fairmot_dla34.pth --output_file=/root/fairmot.onnx

四、日志信息：
  %700 : Float(1, 256, 152, 272) = onnx::Relu(%699) # /usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/functional.py:1061:0
  %wh : Float(1, 4, 152, 272) = onnx::Conv[dilations=[1, 1], group=1, kernel_shape=[1, 1], pads=[0, 0, 0, 0], strides=[1, 1]](%700, %wh.2.weight, %wh.2.bias) # /usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/conv.py:346:0
  %702 : Float(1, 256, 152, 272) = onnx::Conv[dilations=[1, 1], group=1, kernel_shape=[3, 3], pads=[1, 1, 1, 1], strides=[1, 1]](%695, %id.0.weight, %id.0.bias) # /usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/conv.py:346:0
  %703 : Float(1, 256, 152, 272) = onnx::Relu(%702) # /usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/functional.py:1061:0
  %id : Float(1, 128, 152, 272) = onnx::Conv[dilations=[1, 1], group=1, kernel_shape=[1, 1], pads=[0, 0, 0, 0], strides=[1, 1]](%703, %id.2.weight, %id.2.bias) # /usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/conv.py:346:0
  %705 : Float(1, 256, 152, 272) = onnx::Conv[dilations=[1, 1], group=1, kernel_shape=[3, 3], pads=[1, 1, 1, 1], strides=[1, 1]](%695, %reg.0.weight, %reg.0.bias) # /usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/conv.py:346:0
  %706 : Float(1, 256, 152, 272) = onnx::Relu(%705) # /usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/functional.py:1061:0
  %reg : Float(1, 2, 152, 272) = onnx::Conv[dilations=[1, 1], group=1, kernel_shape=[1, 1], pads=[0, 0, 0, 0], strides=[1, 1]](%706, %reg.2.weight, %reg.2.bias) # /usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/conv.py:346:0
  return (%hm, %wh, %id, %reg)

Traceback (most recent call last):
  File "/root/ModelZoo-PyTorch-master/ACL_PyTorch/contrib/cv/tracking/FairMOT/fairmot_pth2onnx.py", line 56, in <module>
    pth2onnx(args.input_file, args.output_file)
  File "/root/ModelZoo-PyTorch-master/ACL_PyTorch/contrib/cv/tracking/FairMOT/fairmot_pth2onnx.py", line 47, in pth2onnx
    enable_onnx_checker=False)
  File "/usr/local/python3.7.5/lib/python3.7/site-packages/torch/onnx/__init__.py", line 168, in export
    custom_opsets, enable_onnx_checker, use_external_data_format)
  File "/usr/local/python3.7.5/lib/python3.7/site-packages/torch/onnx/utils.py", line 69, in export
    use_external_data_format=use_external_data_format)
  File "/usr/local/python3.7.5/lib/python3.7/site-packages/torch/onnx/utils.py", line 503, in _export
    val_add_node_names, val_use_external_data_format, model_file_location)
RuntimeError: ONNX export failed: Couldn't export Python operator _DCNv2

Defined at:
/root/DCNv2/dcn_v2.py(128): forward
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/module.py(676): _slow_forward
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/module.py(690): __call__
/usr/local/python3.7.5/lib/python3.7/site-packages/lib/models/networks/pose_dla_dcn.py(355): forward
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/module.py(676): _slow_forward
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/module.py(690): __call__
/usr/local/python3.7.5/lib/python3.7/site-packages/lib/models/networks/pose_dla_dcn.py(384): forward
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/module.py(676): _slow_forward
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/module.py(690): __call__
/usr/local/python3.7.5/lib/python3.7/site-packages/lib/models/networks/pose_dla_dcn.py(411): forward
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/module.py(676): _slow_forward
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/module.py(690): __call__
/usr/local/python3.7.5/lib/python3.7/site-packages/lib/models/networks/pose_dla_dcn.py(472): forward
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/module.py(676): _slow_forward
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/module.py(690): __call__
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/jit/__init__.py(348): wrapper
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/jit/__init__.py(361): forward
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/nn/modules/module.py(692): __call__
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/jit/__init__.py(278): _get_trace_graph
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/onnx/utils.py(291): _trace_and_get_graph_from_model
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/onnx/utils.py(334): _model_to_graph
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/onnx/utils.py(488): _export
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/onnx/utils.py(69): export
/usr/local/python3.7.5/lib/python3.7/site-packages/torch/onnx/__init__.py(168): export
/root/ModelZoo-PyTorch-master/ACL_PyTorch/contrib/cv/tracking/FairMOT/fairmot_pth2onnx.py(47): pth2onnx
/root/ModelZoo-PyTorch-master/ACL_PyTorch/contrib/cv/tracking/FairMOT/fairmot_pth2onnx.py(56): <module>

获取方法请参考wiki：
https://gitee.com/ascend/modelzoo/wikis/%E5%A6%82%E4%BD%95%E8%8E%B7%E5%8F%96%E6%97%A5%E5%BF%97%E5%92%8C%E8%AE%A1%E7%AE%97%E5%9B%BE?sort_id=4097825

Ascend/modelzoo
暂停

内容风险标识

评论 (2)

Ascend/modelzoo暂停 .gitee-modal { width: 500px !important; }

内容风险标识