EasyCV/easycv/models/loss
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 Yolox improve with REPConv/ASFF/TOOD (#154) 2022-08-24 18:11:15 +08:00
cross_entropy_loss.py refactor detection, support detr and dab-detr (#104) 2022-07-11 14:52:49 +08:00
focal_loss.py Yolox improve with REPConv/ASFF/TOOD (#154) 2022-08-24 18:11:15 +08:00
iou_loss.py Yolox improve with REPConv/ASFF/TOOD (#154) 2022-08-24 18:11:15 +08:00
mse_loss.py initial commit 2022-04-02 20:01:06 +08:00
pytorch_metric_learning.py remove useless import and optimize imports (#84) 2022-06-13 10:29:41 +08:00
utils.py refactor detection, support detr and dab-detr (#104) 2022-07-11 14:52:49 +08:00