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
9517bb80ff
add dn_detr algo ( #119 )
...
support dn-detr-r50: 44.39
refactor set_criterion
2022-07-27 15:06:06 +08:00
eeabb4c6ab
refactor det ( #124 )
2022-07-13 12:11:40 +08:00
5fb0a86f69
support fcos ( #100 )
...
support fcos(38.57)
2022-07-12 18:07:02 +08:00
0b472bacf1
refactor detection, support detr and dab-detr ( #104 )
...
refactor detection, support detr and dab-detr
2022-07-11 14:52:49 +08:00
e09572895b
refactor fp16 ( #94 )
...
* refactor fp16 and fix AMPFP16OptimizerHook
2022-06-21 16:16:09 +08:00
5110de7635
remove RPNHeadNorm, adapted in mmlab_utils ( #90 )
...
* remove RPNHeadNorm, adapted in mmlab_utils
2022-06-16 15:53:43 +08:00
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
3af93bdac5
YOLOX End2end & Blade Support ( #66 )
...
* support export with jit.trace and jit.script and end2end
2022-06-08 20:41:15 +08:00
10266f54a7
update quantize.py code version ( #44 )
...
* compression tutorial
* add details of compression tutorial
* add env of compression tutorial
* update quantize.py code version
2022-05-07 16:07:07 +08:00
055deba53e
initial commit
2022-04-02 20:01:06 +08:00
wuziheng
Chen Jiayu
Cathy0908
zouxinyi0625
Xiaohe You
wenmeng.zwm