请问 detectron2 ModulatedDeformConv 如何使用 DCNv2 的插件？

dlunion / tensorRTIntegrate

TensorRT ONNX Plugin、Inference、Compile

463 stars 114 forks source link

请问 detectron2 ModulatedDeformConv 如何使用 DCNv2 的插件？ #22

Open mchl0203 opened 4 years ago

mchl0203 commented 4 years ago

由于使用 detectron2 框架做模型训练，其中的 dcn 层使用的是 detectron2.layers.ModulatedDeformConv。若直接将 ModulatedDeformConv 替换成 DCNv2 进行推理，结果与原始的不一致（数值比原始的结果要大）。请问各位大佬了解 detectron2 ModulatedDeformConv 与 DCNv2 源码实现的区别吗？

dlunion commented 4 years ago

这个事情需要去调研一下，可能有细节上的区别。我有消息通知你