17 Commits

Author SHA1 Message Date
Cathy0908
5c33d9e2f9
support export blade model for Stgcn (#299)
* support blade for stgcn and add unittest
2023-03-06 10:19:39 +08:00
Cathy0908
4a870cf007
support export jit trace and blade model for pose/keypoints & add pose model zoo (#294)
support export jit trace and blade model for pose/keypoints & add pose model zoo
2023-03-02 19:13:35 +08:00
Cathy0908
edfd482718
add mobilenet itag config (#276)
* add mobilenet itag config
2023-02-08 14:09:14 +08:00
Cathy0908
56bf3edf4e
fix export ssl models (#257) 2022-12-12 11:31:07 +08:00
Cathy0908
a36948811f
update yolox template config (#243)
* update yolox config templatee

* update oss sync hook,support sync all export  files

* add sync export model config
2022-12-05 22:01:44 +08:00
tuofeilun
23f2b0e399
Adapt designer (#235)
1. Use original config as startup script. (For details, see refactor config parsing method #225)
2. Refactor the splicing rules of the check_base_cfg_path function in the EasyCV/easycv/utils/config_tools.py
3. Support three ways to pass class_list parameter.
4. Fix the bug that clsevalutor may make mistakes when evaluating top5.
5. Fix the bug that the distributed export cannot export the model.
6. Fix the bug that the load pretrained model key does not match.
7. support cls data source itag.
2022-12-01 17:47:10 +08:00
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
f19a05a536
fix yolox export error when output is empty (#217)
* fix yolox export error when output is empty
2022-10-28 10:26:53 +08:00
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