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 |
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 |
|
AllentDan
|
9647ab12db
|
Add cls visualization (#18)
* add cls visualization tool
* fix lint
* remove reduandant
* align result
|
2021-07-22 09:58:53 +08:00 |
RunningLeon
|
10a2385d01
|
Support exporting to ONNX for Faster R-CNN (#20)
* update rewriter for rpn head
* new nms implement
* update bbox_head to support fasterrcnn with tensorrt
* rewrite SingleRoIExtractor
* remove unnecessary import
* fix fcos and bbox_nms
* resolve comments
* resolve comments
Co-authored-by: grimoire <yaoqian@sensetime.com>
|
2021-07-21 19:46:23 +08:00 |
|
AllentDan
|
1718501b4f
|
[Feature] Support fsaf fcos (#14)
* add fcos and fsaf
* remove is_in_onnx_export
* fix lint
* fix lint
* remove forwards
|
2021-07-13 20:13:39 +08:00 |
|
RunningLeon
|
c4b7dad2ec
|
add visualization for mmdet (#13)
* add visualization for mmdet
* resolve comments
* update code, enable visualize on nvidia driver>11
Co-authored-by: grimoire <yaoqian@sensetime.com>
|
2021-07-13 17:21:02 +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 |
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 |