search

grimoire / mmdetection-to-tensorrt

convert mmdetection model to tensorrt, support fp16, int8, batch input, dynamic shape etc.
Apache License 2.0
587 stars 85 forks source link

Support for detectron2 #57

Open petoor opened 3 years ago

petoor commented 3 years ago

Hi.

Are there any plans to extend this repo to also transform detectron2 models to tensorrt?

grimoire commented 3 years ago

No, That's an attractive idea, But I'm afraid I do not have enough time to finish it. I plan to add a tutorial about how to add convertor or module support. That might help those who want to create tools for detectron2.