EasyCV/easycv/models/utils
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
..
__init__.py refactor detection, support detr and dab-detr (#104) 2022-07-11 14:52:49 +08:00
activation.py [feature] add segmentation modules (#71) 2022-05-31 20:19:17 +08:00
conv_module.py [feature] add segmentation modules (#71) 2022-05-31 20:19:17 +08:00
conv_ws.py initial commit 2022-04-02 20:01:06 +08:00
dist_utils.py refactor detection, support detr and dab-detr (#104) 2022-07-11 14:52:49 +08:00
gather_layer.py initial commit 2022-04-02 20:01:06 +08:00
init_weights.py initial commit 2022-04-02 20:01:06 +08:00
multi_pooling.py remove useless import and optimize imports (#84) 2022-06-13 10:29:41 +08:00
norm.py initial commit 2022-04-02 20:01:06 +08:00
ops.py Yolox improve with REPConv/ASFF/TOOD (#154) 2022-08-24 18:11:15 +08:00
pos_embed.py initial commit 2022-04-02 20:01:06 +08:00
res_layer.py initial commit 2022-04-02 20:01:06 +08:00
scale.py initial commit 2022-04-02 20:01:06 +08:00
sobel.py initial commit 2022-04-02 20:01:06 +08:00
transformer.py refactor detection, support detr and dab-detr (#104) 2022-07-11 14:52:49 +08:00