9 Commits

Author SHA1 Message Date
Cathy0908
5c33d9e2f9
support export blade model for Stgcn (#299)
* support blade for stgcn and add unittest
2023-03-06 10:19:39 +08:00
Cathy0908
f8c9a9a1c9
Improve the performance of bevformer (#224)
Improve the performance of bevformer

* add hybrid brach (#232)

Co-authored-by: yhq <yanhaiqiang.yhq@alibaba-inc.com>
2022-11-23 21:32:08 +08:00
wuziheng
9aaa600f79
Yolox improve with REPConv/ASFF/TOOD (#154)
* add attention layer and more loss function

* add attention layer and various loss functions

* add siou loss

* add tah,various attention layers, and different loss functions

* add asff sim, gsconv

* blade utils fit faster

* blade optimize for yolox static & fp16

* decode output for yolox control by cfg

* add reparameterize_models for export

* e2e trt_nms plugin export support and numeric test

* split preprocess from end2end+blade, speedup from 17ms->7.2ms

Co-authored-by: zouxinyi0625 <zouxinyi.zxy@alibaba-inc.com>
2022-08-24 18:11:15 +08:00
Chen Jiayu
9517bb80ff
add dn_detr algo (#119)
support dn-detr-r50: 44.39
refactor set_criterion
2022-07-27 15:06:06 +08:00
zouxinyi0625
3af93bdac5
YOLOX End2end & Blade Support (#66)
* support export with jit.trace and jit.script and end2end
2022-06-08 20:41:15 +08:00
Cathy0908
ecbfbbb359
fix unittest (#57)
* fix unittest and skip quantize unittest
2022-05-16 15:02:12 +08:00
wenmeng.zwm
eaae92a4a5 remove comment 2022-04-21 16:07:59 +08:00
wenmeng.zwm
7534a80195 add unittest and fix another bug using cls_head with_avg_pool=True 2022-04-21 16:07:33 +08:00
wenmeng.zwm
055deba53e initial commit 2022-04-02 20:01:06 +08:00