search

CaoWGG / TensorRT-CenterNet

tensorrt5 , centernet , centerface, deform conv, int8
767 stars 159 forks source link

转onnx时报错：RuntimeError: ONNX export failed: Couldn't export Python operator DeformConvFunction #96

Open myl980 opened 3 years ago

myl980 commented 3 years ago

在编译好新的dcn过后，转onnx时报错： File "/home/fwq/anaconda3/envs/py35/lib/python3.5/site-packages/torch/onnx/init.py", line 132, in export strip_doc_string, dynamic_axes) File "/home/fwq/anaconda3/envs/py35/lib/python3.5/site-packages/torch/onnx/utils.py", line 64, in export example_outputs=example_outputs, strip_doc_string=strip_doc_string, dynamic_axes=dynamic_axes) File "/home/fwq/anaconda3/envs/py35/lib/python3.5/site-packages/torch/onnx/utils.py", line 340, in _export params_dict, opset_version, dynamic_axes, defer_weight_export, operator_export_type, strip_doc_string) RuntimeError: ONNX export failed: Couldn't export Python operator DeformConvFunction 请问这种问题要怎么解决？

jedibobo commented 3 years ago

同问

ZouYunzhe commented 3 years ago

@myl353 请问问题解决了吗？