Commit Graph

9 Commits (f8c9a9a1c958b17554827f0bac33cab069b72735)

Author SHA1 Message Date
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
zouxinyi0625 9ad1a8f448
Fix YOLOX-Edge problem and update yolox-pai turtorial (#182)
* complete yolox introduction

* fix env problem with nni2.6.1

* fix yolox-edge problem

*  add docker turtorial

* add blade test

* support old export yolox model

Co-authored-by: wuziheng <zhoulou.wzh@alibaba-inc.com>
2022-09-06 17:11:12 +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