AllentDan
8fe8056080
[Feature] Add trt ort mdcn plugins ( #43 )
...
* add trt mdcn plugin
* add onnxruntime mdcn op
* add mdcn trt ort plugins
* fix lint
* remove comment
* remove plugin condition lines
* apply new form
* use serialized_size
2021-08-25 10:17:21 +08:00
RunningLeon
90ce7207da
[Feature] Support Mask R-CNN with ONNX Runtime and TensorRT ( #24 )
...
* add grid_sampler plugin
* update test_mixins to support maskrcnn
* add config for maskrcnn exporting to onnx
* fix lints
* fix lint
* resolve comments
* fix lint
* fix lint
* fix lint
* support trt8
* move mark to single_stage
* fix lint
* remove mark in single stage
2021-07-28 19:02:43 +08:00
grimoire
53916af4e3
fix multilevel roi align on trt8
2021-07-28 18:45:21 +08:00
q.yao
6ff8e96e71
add TRT8 support ( #23 )
2021-07-28 11:27:07 +08:00
AllentDan
b9e64f9e1c
Add trt instance norm plugin ( #16 )
...
* add trt instance norm plugin
* last line empty
* fix clang format
* fix grid_sample clang format
* remove redundant
* fix lint
* refine codes
* fix clang format
* clang format
* clang format
* clang format
2021-07-16 16:05:09 +08:00
q.yao
3fa94f4424
[Feature] better tensorrt cpp code ( #11 )
...
* better tensorrt cpp code
* fix end of file
2021-07-12 10:01:52 +08:00
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
