Commit Graph

10 Commits (f8c9a9a1c958b17554827f0bac33cab069b72735)

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
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
tuofeilun 9f01a37ad4
Refactor ViTDet backbone and simple feature pyramid (#177)
1. The vitdet backbone implemented by d2 is about 20% faster than the vitdet backbone originally reproduced by easycv.
2. 50.57 -> 50.65
2022-09-16 11:03:53 +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
tuofeilun 6821cdc9db
refactor detection predictor (#158)
The use of unified DetrPredictor for detection test and predictor.
2022-08-26 10:28:17 +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 9424cc9660
vitdet visualize and predictor (#102)
*vitdet bugfix

* support vitdet_predictor
2022-06-30 09:35:27 +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 055deba53e initial commit 2022-04-02 20:01:06 +08:00