208 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
66a099faf9
[Feature] Apis unit test (#7)
* add apis test

* split torch2onnx impl, prepare for codebase test

* add is_available to backend

* lint
2021-07-05 12:51:43 +08:00
grimoire
5b2459570a better register 2021-06-16 19:25:09 +08:00
grimoire
100ba694ce add init 2021-06-16 17:56:17 +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
grimoire
f90ebf8c2c add register 2021-06-11 13:26:05 +08:00