diff --git a/docs/en/03-benchmark/benchmark.md b/docs/en/03-benchmark/benchmark.md index b45eaff39..1d0b61782 100644 --- a/docs/en/03-benchmark/benchmark.md +++ b/docs/en/03-benchmark/benchmark.md @@ -625,6 +625,28 @@ Users can directly test the performance through [how_to_evaluate_a_model.md](../ - - - + + EfficientFormer + top-1 + 80.46 + 80.45 + 80.46 + 80.46 + - + - + - + - + + + top-5 + 94.99 + 94.98 + 94.99 + 94.99 + - + - + - + - diff --git a/docs/en/04-supported-codebases/mmpretrain.md b/docs/en/04-supported-codebases/mmpretrain.md index fecae8d0d..feb568bfa 100644 --- a/docs/en/04-supported-codebases/mmpretrain.md +++ b/docs/en/04-supported-codebases/mmpretrain.md @@ -185,3 +185,4 @@ Besides python API, mmdeploy SDK also provides other FFI (Foreign Function Inter | [MobileOne](https://github.com/open-mmlab/mmpretrain/tree/main/configs/mobileone) | Y | Y | N | N | ? | N | | [EfficientNet](https://github.com/open-mmlab/mmpretrain/tree/main/configs/efficientnet) | Y | Y | N | N | ? | N | | [Conformer](https://github.com/open-mmlab/mmpretrain/tree/main/configs/conformer) | Y | Y | N | N | ? | N | +| [EfficientFormer](https://github.com/open-mmlab/mmpretrain/tree/main/configs/efficientformer) | Y | Y | Y | N | ? | Y | diff --git a/docs/zh_cn/03-benchmark/benchmark.md b/docs/zh_cn/03-benchmark/benchmark.md index c9e7589a6..1f0f5429d 100644 --- a/docs/zh_cn/03-benchmark/benchmark.md +++ b/docs/zh_cn/03-benchmark/benchmark.md @@ -608,6 +608,7 @@ GPU: ncnn, TensorRT, PPLNN 81.18 - - + - top-5 @@ -618,6 +619,30 @@ GPU: ncnn, TensorRT, PPLNN 95.61 - - + - + + EfficientFormer + top-1 + 80.46 + 80.45 + 80.46 + 80.46 + - + - + - + - + + + top-5 + 94.99 + 94.98 + 94.99 + 94.99 + - + - + - + - + diff --git a/docs/zh_cn/04-supported-codebases/mmpretrain.md b/docs/zh_cn/04-supported-codebases/mmpretrain.md index d759713ce..5bd8d6487 100644 --- a/docs/zh_cn/04-supported-codebases/mmpretrain.md +++ b/docs/zh_cn/04-supported-codebases/mmpretrain.md @@ -190,3 +190,4 @@ for label_id, score in result: | [MobileOne](https://github.com/open-mmlab/mmpretrain/tree/main/configs/mobileone) | Y | Y | N | N | ? | N | | [EfficientNet](https://github.com/open-mmlab/mmpretrain/tree/main/configs/efficientnet) | Y | Y | N | N | ? | N | | [Conformer](https://github.com/open-mmlab/mmpretrain/tree/main/configs/conformer) | Y | Y | N | N | ? | N | +| [EfficientFormer](https://github.com/open-mmlab/mmpretrain/tree/main/configs/efficientformer) | Y | Y | Y | N | ? | Y | diff --git a/mmdeploy/codebase/mmpretrain/deploy/classification.py b/mmdeploy/codebase/mmpretrain/deploy/classification.py index 50849583f..fb36e0cd3 100644 --- a/mmdeploy/codebase/mmpretrain/deploy/classification.py +++ b/mmdeploy/codebase/mmpretrain/deploy/classification.py @@ -332,6 +332,9 @@ class Classification(BaseTask): dict: Composed of the postprocess information. """ postprocess = self.model_cfg.model.head + if postprocess['type'] == 'EfficientFormerClsHead': + postprocess['type'] = 'LinearClsHead' + if 'topk' not in postprocess: topk = (1, ) logger = get_root_logger() diff --git a/tests/regression/mmpretrain.yml b/tests/regression/mmpretrain.yml index c479ae4a0..c11f4fb67 100644 --- a/tests/regression/mmpretrain.yml +++ b/tests/regression/mmpretrain.yml @@ -237,3 +237,13 @@ models: pipelines: - *pipeline_ort_dynamic_fp32 - *pipeline_trt_dynamic_fp32 + + - name: EfficientFormer + metafile: configs/efficientformer/metafile.yml + model_configs: + - configs/efficientformer/efficientformer-l1_8xb128_in1k.py + pipelines: + - *pipeline_ts_fp32 + - *pipeline_ort_dynamic_fp32 + - *pipeline_trt_dynamic_fp32 + - *pipeline_openvino_dynamic_fp32