EasyCV/easycv/models/backbones
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
benchmark_mlp.py Expand cls model zoo (#55) 2022-06-01 11:01:29 +08:00
bninception.py bugfix_swintiny (#79) 2022-06-06 12:03:05 +08:00
conv_mae_vit.py add detection benchmark model ConvViTDet for FastConvMAE (#121) 2022-07-18 17:11:41 +08:00
conv_vitdet.py add detection benchmark model ConvViTDet for FastConvMAE (#121) 2022-07-18 17:11:41 +08:00
darknet.py Yolox improve with REPConv/ASFF/TOOD (#154) 2022-08-24 18:11:15 +08:00
efficientformer.py support Efficientformer algo (#128) 2022-07-27 11:36:03 +08:00
genet.py bugfix_swintiny (#79) 2022-06-06 12:03:05 +08:00
hrnet.py bugfix_swintiny (#79) 2022-06-06 12:03:05 +08:00
inceptionv3.py bugfix_swintiny (#79) 2022-06-06 12:03:05 +08:00
lighthrnet.py bugfix_swintiny (#79) 2022-06-06 12:03:05 +08:00
mae_vit_transformer.py add Fast ConvMAE (#101) 2022-07-08 12:11:37 +08:00
mnasnet.py bugfix_swintiny (#79) 2022-06-06 12:03:05 +08:00
mobilenetv2.py bugfix_swintiny (#79) 2022-06-06 12:03:05 +08:00
network_blocks.py Yolox improve with REPConv/ASFF/TOOD (#154) 2022-08-24 18:11:15 +08:00
pytorch_image_models_wrapper.py Swint_cifar config bugfix (#95) 2022-06-16 14:46:47 +08:00
repvgg_yolox_backbone.py Yolox improve with REPConv/ASFF/TOOD (#154) 2022-08-24 18:11:15 +08:00
resnest.py bugfix_swintiny (#79) 2022-06-06 12:03:05 +08:00
resnet.py bugfix_swintiny (#79) 2022-06-06 12:03:05 +08:00
resnet_jit.py bugfix_swintiny (#79) 2022-06-06 12:03:05 +08:00
resnext.py bugfix_swintiny (#79) 2022-06-06 12:03:05 +08:00
shuffle_transformer.py bugfix_swintiny (#79) 2022-06-06 12:03:05 +08:00
swin_transformer_dynamic.py remove useless import and optimize imports (#84) 2022-06-13 10:29:41 +08:00
vit_transfomer_dynamic.py add detection benchmark model ConvViTDet for FastConvMAE (#121) 2022-07-18 17:11:41 +08:00
vitdet.py remove RPNHeadNorm, adapted in mmlab_utils (#90) 2022-06-16 15:53:43 +08:00
xcit_transformer.py bugfix_swintiny (#79) 2022-06-06 12:03:05 +08:00