AllentDan
f607f1965b
[Feature] Align datasets ( #29 )
...
* add test tool and re-orgnize apis.utils
* handle topk and refine codes
* add cls export and test support
* fix lint
* move ort into wrapper
* resolve conflicts
* resolve comments
* resolve conflicts
* resolve comments and padding mrcnn
* resolve comments
2021-08-03 17:12:44 +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
q.yao
ff39377833
[Feature] Retinanet split support ( #19 )
...
* prepare for split
* mark multiclass nms
* redesign mark decorator to handle input/output names better (#17 )
* redesign mark decorator to handle input/output names better
* fixed a bug in rename_value; robustify extract_model
* fix lint
* fix log in deploy pipeline
* fix onnxruntime config
* fix lint
* update connect init
* rename is_inspecting
* simplify function_mark
Co-authored-by: Li Zhang <lzhang329@gmail.com>
2021-07-23 13:18:32 +08:00
RunningLeon
9f0e2d4339
changed codebase structure ( #10 )
...
* add delta2bbox rewriter
* rename onnx2trt
* add rewriter: anchor_generator_single_level_grid_priors
* Revert "add rewriter: anchor_generator_single_level_grid_priors"
This reverts commit ac7cf272942c4787bf143c0d67e414b0d2603b75.
* update comments
* remove clamp rewriter
* remove unused func
* reconstruct codes
* remomve mark
* create export package for codebases
* remove grid_sampler symbolic
* reorganize nms parts
* fix conflicts
* update unittest
2021-07-12 16:26:44 +08:00
lzhangzz
66300c0c74
Supports setting input/output names explicitly via mark decorator ( #4 )
...
* setting names of inputs and outputs via mark decorator
* fix bug in mark, move extract_model to mmdeploy/apis and add test cases
* remove fsaf rewrite, which is not needed any more after recent changes
* replace print with log and add fixture to clear test tmp file
2021-07-07 14:15:26 +08:00
q.yao
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
q.yao
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
grimoire
e08dbf26b3
add CI
2021-06-29 10:10:15 +08:00
lzhangzz
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
q.yao
6c47ee3d2a
add tensorrt support ( #2 )
2021-06-23 13:14:28 +08:00
grimoire
2bd96243e2
better mmdet support
2021-06-18 15:16:21 +08:00
maningsheng
2317ee659a
support retinanet pytorch2onnx
2021-06-17 17:37:08 +08:00
grimoire
12792727e7
add configs, add convert pipeline
2021-06-17 15:28:23 +08:00
