10 Commits

Author SHA1 Message Date
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
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
maningsheng
2317ee659a support retinanet pytorch2onnx 2021-06-17 17:37:08 +08:00
grimoire
b777d27bcf add return value in deploy.py 2021-06-17 17:26:32 +08:00
grimoire
3bf3b9cdda fix lint 2021-06-17 15:29:12 +08:00
grimoire
12792727e7 add configs, add convert pipeline 2021-06-17 15:28:23 +08:00