52fd08febd
[Feature] much better retinanet ( #6 )
...
* better retinanet support
* prepare split export tensorrt
* optimizer cfg
* free anchor when static shape
* fix docstring
* use function rewriter instead of module rewriter on retinanet
* fix bug of mmcls tensorrt config
* add single stage mark, static shape support
2021-07-01 17:32:33 +08:00
27880afdcd
[Feature] better retinanet support ( #5 )
...
* better retinanet support
* prepare split export tensorrt
* optimizer cfg
* free anchor when static shape
* fix docstring
2021-07-01 11:42:07 +08:00
8fd4ef4d47
remove unused flag, fix lint
2021-06-29 10:23:52 +08:00
e08dbf26b3
add CI
2021-06-29 10:10:15 +08:00
ef41f69553
add model splitting support ( #1 )
...
* add function marker and model extractor
* add fsaf split & partial mask rcnn split, import extract.py
* 1. add value renaming 2. add apply_marks in config to turn on/off marks
* rewind changes on pytorch2onnx.py
Co-authored-by: q.yao <streetyao@live.com>
2021-06-28 20:35:15 +08:00
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
6c47ee3d2a
add tensorrt support ( #2 )
2021-06-23 13:14:28 +08:00
6eb2e89016
update cmakelists
2021-06-21 10:34:48 +08:00
2bd96243e2
better mmdet support
2021-06-18 15:16:21 +08:00
2317ee659a
support retinanet pytorch2onnx
2021-06-17 17:37:08 +08:00
b777d27bcf
add return value in deploy.py
2021-06-17 17:26:32 +08:00
34caea27c1
use function rewriter convert classifier
2021-06-17 16:04:45 +08:00
3bf3b9cdda
fix lint
2021-06-17 15:29:12 +08:00
12792727e7
add configs, add convert pipeline
2021-06-17 15:28:23 +08:00
5b2459570a
better register
2021-06-16 19:25:09 +08:00
100ba694ce
add init
2021-06-16 17:56:17 +08:00
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
1e47821e49
add backend plugin build system
2021-06-15 20:44:45 +08:00
f90ebf8c2c
add register
2021-06-11 13:26:05 +08:00
b6d8d48d7c
Initial commit
2021-06-11 13:24:18 +08:00
q.yao
grimoire
lzhangzz
maningsheng