Unsure how to apply DCNv2 Plugin to onnx-tensorrt
Opened this issue · 2 comments
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?
Did you solve it?
pytorch1.1 pytorch1.4 works fine ,
pytorch1.5.1 and above are failed