CaoWGG / TensorRT-CenterNet

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

Register custom Op 'DCNv2' #94

Open HardysJin opened 3 years ago

HardysJin commented 3 years ago

Hi,

I can successfully convert the CenterNet to onnx but fail to run/verify the onnx model using, for example

onnx.checker.check_model("CenterNet.onnx")

The error:

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

The custom operator (DCNv2) cannot be recognized by onnx. Any idea how to register DCNv2 in onnx?