q.yao
|
fd40119694
|
[Feature] TensorRT int8 support (#35)
* fix custom ops support, fix multiple mark bug, add name mapping
* check if the value_info need to be added
* remove unnecessary print
* add nms implement
* two stage split wip
* add two stage split
* add split retinanet visualize
* add two stage split (wip)
* finish two stage split
* fix lint
* move parse string to mmdeploy.utils
* add func mark count dict
* use assert_cfg_valid
* update func count before add Mark
* fix dynamic shape support
* add calib data generator
* create calib dataset
* finish end2end int8
* add split two stage tensorrt visualize
|
2021-08-19 12:56:00 +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 |
|
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 |
|
q.yao
|
6c47ee3d2a
|
add tensorrt support (#2)
|
2021-06-23 13:14:28 +08:00 |
|
grimoire
|
12792727e7
|
add configs, add convert pipeline
|
2021-06-17 15:28:23 +08:00 |
|