Commit Graph

3 Commits (a36e0e32a4c67a26ed2f0c1e41e021d5083cc7df)

Author SHA1 Message Date
zouxinyi0625 8a484e88fd
Add yolox jit script model end2end support (#215)
* add yolox jit script model end2end support

* update export.md
2022-10-26 15:09:05 +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
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