9 Commits

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
yhq	bb53e066be	fix missing vit model (#197) ... * fix missing vit model * set pretrained false while export cls model	2022-09-19 19:52:21 +08:00
Cathy0908	5ac6381758	add error code (#146) ... * add error code	2022-09-19 16:07:04 +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
Cathy0908	4ed4974eab	update config template and fix some bugs (#134) ... * update yolox config template and fix bugs	2022-07-27 10:25:37 +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	7534a80195	add unittest and fix another bug using cls_head with_avg_pool=True	2022-04-21 16:07:33 +08:00
wenmeng.zwm	2a7eac471d	fix cls predictor error	2022-04-21 11:53:44 +08:00
wenmeng.zwm	055deba53e	initial commit	2022-04-02 20:01:06 +08:00