mirrors
/
mmdeploy
mirror of https://github.com/open-mmlab/mmdeploy.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Support MobileOne on ONNXRuntime and TensorRT (#1268) 

* Support MobileOne

* Updated supported model readme (Add MobileOne)

* Update supported_models.md

* Updated supported model readme (Add MobileOne)

* Update mmcls.yml

* Support MobileOne

* Remove other files

* Resolve lint error

* Remove converted model files

* Remove json files

* Update mmcls.yml

* Update mmcls.yml

* Update benchmark doc

Co-authored-by: SenseTime Research Singapore <SENSETIME\research.sgres@sg0016000001u.domain.sensetime.com>
pull/1372/head
SsTtOoNnEe 2022-11-14 17:46:42 +08:00 committed by GitHub
parent 83756b97c6
commit 680a3fb239
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docs/en/03-benchmark/supported_models.md
View File

@ -30,6 +30,7 @@ The table below lists the models that are guaranteed to be exportable to other b
| [ShuffleNetV2](https://github.com/open-mmlab/mmclassification/tree/1.x/configs/shufflenet_v2) | MMClassification | Y | Y | Y | Y | Y | Y | Y | Y |
| [VisionTransformer](https://github.com/open-mmlab/mmclassification/tree/1.x/configs/vision_transformer) | MMClassification | Y | Y | Y | Y | ? | Y | Y | N |
| [SwinTransformer](https://github.com/open-mmlab/mmclassification/tree/1.x/configs/swin_transformer) | MMClassification | Y | Y | Y | N | ? | N | ? | N |
| [MobileOne](https://github.com/open-mmlab/mmclassification/tree/1.x/configs/mobileone) | MMClassification | N | Y | Y | N | N | N | N | N |
| [FCN](https://github.com/open-mmlab/mmsegmentation/tree/1.x/configs/fcn) | MMSegmentation | Y | Y | Y | Y | Y | Y | Y | Y |
| [PSPNet](https://github.com/open-mmlab/mmsegmentation/tree/1.x/configs/pspnet)[\*static](#note) | MMSegmentation | Y | Y | Y | Y | Y | Y | Y | Y |
| [DeepLabV3](https://github.com/open-mmlab/mmsegmentation/tree/1.x/configs/deeplabv3) | MMSegmentation | Y | Y | Y | Y | Y | Y | Y | N |

1
docs/zh_cn/03-benchmark/supported_models.md
View File

@ -30,6 +30,7 @@
| [ShuffleNetV2](https://github.com/open-mmlab/mmclassification/tree/1.x/configs/shufflenet_v2) | MMClassification | Y | Y | Y | Y | Y | Y | Y | Y |
| [VisionTransformer](https://github.com/open-mmlab/mmclassification/tree/1.x/configs/vision_transformer) | MMClassification | Y | Y | Y | Y | ? | Y | Y | N |
| [SwinTransformer](https://github.com/open-mmlab/mmclassification/tree/1.x/configs/swin_transformer) | MMClassification | Y | Y | Y | N | ? | N | ? | N |
| [MobileOne](https://github.com/open-mmlab/mmclassification/tree/1.x/configs/mobileone) | MMClassification | N | Y | Y | N | N | N | N | N |
| [FCN](https://github.com/open-mmlab/mmsegmentation/tree/1.x/configs/fcn) | MMSegmentation | Y | Y | Y | Y | Y | Y | Y | Y |
| [PSPNet](https://github.com/open-mmlab/mmsegmentation/tree/1.x/configs/pspnet)[\*static](#note) | MMSegmentation | Y | Y | Y | Y | Y | Y | Y | Y |
| [DeepLabV3](https://github.com/open-mmlab/mmsegmentation/tree/1.x/configs/deeplabv3) | MMSegmentation | Y | Y | Y | Y | Y | Y | Y | N |

8
tests/regression/mmcls.yml
View File

@ -212,6 +212,14 @@ models:
- *pipeline_ort_dynamic_fp32
- *pipeline_trt_static_fp16
- name: MobileOne
metafile: configs/mobileone/metafile.yml
model_configs:
- configs/mobileone/mobileone-s0_8xb128_in1k.py
pipelines:
- *pipeline_trt_static_fp16
- *pipeline_ort_dynamic_fp32
- name: EfficientNet
metafile: configs/efficientnet/metafile.yml
model_configs: