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 |