Commit Graph

5 Commits (893d879b4471b9bddb3b936211c9249e5d78a3f1)

Author SHA1 Message Date
Cathy0908 7a89d1b7b8
refactor yolox predictor, support batch inference (#198)
* update yolox predictor
2022-09-23 13:51:06 +08:00
Cathy0908 5dfe7b2898
update some predcitors, support batch inference (#195)
update some predcitors, support batch inference
2022-09-20 10:04:42 +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
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