7 Commits

Author SHA1 Message Date
q.yao
8408ae2bcf
[Feature] Multi level roi align (#9)
* add multi level roi align

* better trt plugin code

* fix lint

* optimizer trt_multi_level_roi_align.cpp

* fix lint
2021-07-09 13:56:50 +08:00
grimoire
8fd4ef4d47 remove unused flag, fix lint 2021-06-29 10:23:52 +08:00
grimoire
e08dbf26b3 add CI 2021-06-29 10:10:15 +08:00
q.yao
5998d24766
[Feature] Add TensorRT batched NMS support (#3)
* add trt batched_nms plugin

* update trt batched nms
2021-06-25 19:31:16 +08:00
grimoire
6eb2e89016 update cmakelists 2021-06-21 10:34:48 +08:00
grimoire
93408c7cd8 add type annotation
add onnxruntime plugin support, add more custom ops

update function_rewriter, enable use origin function

add setup.cfg

update setup.cfg
2021-06-16 15:36:58 +08:00
grimoire
1e47821e49 add backend plugin build system 2021-06-15 20:44:45 +08:00