Commit Graph

8 Commits (8c3ba59aaf189b5a7e6ce88763f2d9acdf90a3cc)

Author SHA1 Message Date
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
Chen Jiayu e61488d319
Support ViTDet algo (#35)
* adapt mmlab modules
* add vitdet
* support conv aggregation
* modify vitdet load pretrained
* support fp16
* modify defaultformatbundle
* modify aug
* bugfix sampler
* bugfix mmresize
* bugfix fp16&nonetype
* bugfix filterannotation
* support dlc
* bugfix dist
* bugfix detsourcecoco
* smodify mmdet_parse_losses
* bugfix nan
* bugfix eval
* bugfix data=nonetype
* modify resize_embed
* support vitdet_conv
* add vitdet_conv init_weight
* add test_vitdet
* uniform rand_another
* uniform use fp16 method
* add test_fp16

Co-authored-by: jiangnana.jnn <jiangnana.jnn@alibaba-inc.com>
2022-06-10 21:49:32 +08:00
Cathy0908 d76d5d79fc
set multiprocess env for speedup (#77) 2022-06-01 10:16:02 +08:00
Cathy0908 00f2e2eb64
fix eval metric during training and add metric unittest (#67)
* fix eval metric during training and add metric unittest
2022-05-19 16:07:36 +08:00
Cathy0908 52c2a35784
fix eval.py, remove oss_config from build_dataloader (#62) 2022-05-16 20:21:18 +08:00
Cathy0908 b5fb2b70c7
[feature]: support mmdet models config (#25)
* support mmdet models

* add mmlab_models_usage_guide.md

* remove tools/test.py
2022-05-11 17:44:06 +08:00
wenmeng.zwm 055deba53e initial commit 2022-04-02 20:01:06 +08:00