search

dlunion / tensorRTIntegrate

TensorRT ONNX Plugin、Inference、Compile
463 stars 114 forks source link

Unsure how to apply DCNv2 Plugin to onnx-tensorrt #30

Open austinmw opened 4 years ago

austinmw commented 4 years ago

Hi, thanks very much for creating this repository!

I've exported CenterTrack with DCNv2 to ONNX with your replacements, but am not sure how to add DCNv2 plugin for onnx-tensorrt parser.

I get the expected error:

onnx.onnx_cpp2py_export.checker.ValidationError: No Op registered for Plugin with domain_version of 9

==> Context: Bad node spec: input: "561" input: "562" input: "dla_up.ida_0.proj_1.conv.weight" input: "dla_up.ida_0.proj_1.conv.bias" output: "563" op_type: "Plugin" attribute { name: "info" s: "{\"dilation\": [1, 1], \"padding\": [1, 1], \"stride\": [1, 1], \"deformable_groups\": 1}" type: STRING } attribute { name: "name" s: "DCNv2" type: STRING }

If you find time, would you mind showing how I should modify my onnx-tensorrt install to add the plugin?

luyao-cv commented 4 years ago

Did you solve it?

KangolHsu commented 3 years ago

pytorch1.1 pytorch1.4 works fine ,

pytorch1.5.1 and above are failed