q.yao
ea3078aa56
Update custom ops namespace ( #158 )
...
* Update custom ops namespace
* rename register, rename generated library name
* rename domain to mmcv
2021-11-01 10:48:21 +08:00
q.yao
b5797c9e6c
Update c++14 ( #149 )
2021-10-26 10:54:45 +08:00
lvhan028
6fdf6b8616
apply cmake-format and refactor cmake scripts of building onnx and te… ( #99 )
...
* apply cmake-format and refactor cmake scripts of building onnx and tensorrt ops
* add static target
* suppress 'CMAKE_CUDA_ARCHITECTURES' warning when cmake with version 18 or later is used
* fix typo
2021-10-13 16:42:11 +08:00
q.yao
f56a30025a
[Enhance] Add spell hook ( #116 )
...
* add spell hook
* fix code spell
2021-10-09 11:34:14 +08:00
q.yao
a335444a49
[Fix]: Remove TensorRT NMS plugin ( #113 )
...
* Remove TensorRT NMS plugin
* remove nms test
2021-10-09 11:14:38 +08:00
q.yao
be2df638b1
less deprecated warning when compile tensorrt ops ( #103 )
2021-09-29 10:49:44 +08:00
q.yao
067b915383
fix void pointer arithmetic ( #98 )
2021-09-27 10:52:26 +08:00
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
