search

lucasjinreal / DCNv2_latest

DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)
BSD 3-Clause "New" or "Revised" License
626 stars 124 forks source link

Export to ONNX #59

Open abhiagwl4262 opened 2 years ago

abhiagwl4262 commented 2 years ago

hey @lbin @CharlesShang, Thanks for the work. Can you please add a script which is able to export _DCN_v2 to ONNX and run inference with onnxruntime ?

hitbuyi commented 2 years ago

I also hope such kind of job related to deployment on TensorRT

abhiagwl4262 commented 2 years ago

@hitbuyi I have successfully export DCNv2 to ONNX and TensorRT. The trick is - use mmcv's ModulatedDeformConv2D instead of this repo.

Check this for more details - https://medium.com/@abhiagwl4262/yolact-pyotorch-to-onnx-tensorrt-9e6a37f6b974

hitbuyi commented 2 years ago

@hitbuyi I have successfully export DCNv2 to ONNX and TensorRT. The trick is - use mmcv's ModulatedDeformConv2D instead of this repo.

Check this for more details - https://medium.com/@abhiagwl4262/yolact-pyotorch-to-onnx-tensorrt-9e6a37f6b974

@abhiagwl4262 Thanks for your useful information My model is SMOKE, which uses another version of DCNv2, is it possible to use mmcv's DCNv2 to replace it?