Commit Graph

7 Commits (2cb2bded7dc8816188707d93f8100dc440e73f4a)

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
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
Chen Jiayu 5fb0a86f69
support fcos (#100)
support fcos(38.57)
2022-07-12 18:07:02 +08:00
Chen Jiayu 0b472bacf1
refactor detection, support detr and dab-detr (#104)
refactor detection, support detr and dab-detr
2022-07-11 14:52:49 +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
Xiaohe You 10266f54a7
update quantize.py code version (#44)
* compression tutorial

* add details of compression tutorial

* add env of compression tutorial

* update quantize.py code version
2022-05-07 16:07:07 +08:00
wenmeng.zwm 055deba53e initial commit 2022-04-02 20:01:06 +08:00