Commit Graph

6 Commits (b198c5a81fbe6d93bdf1c88ead17fd6d584781e4)

Author SHA1 Message Date
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
Cathy0908 4ed4974eab
update config template and fix some bugs (#134)
* update yolox config template and fix bugs
2022-07-27 10:25:37 +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
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 2a7eac471d fix cls predictor error 2022-04-21 11:53:44 +08:00
wenmeng.zwm 055deba53e initial commit 2022-04-02 20:01:06 +08:00