mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
0111e1f10f
* add swin for cls * add ut and doc * reduce trt batch size * add regression test * resolve comments * remove useless rewriting logic
2.1 KiB
2.1 KiB
mmcls 模型支持列表
MMClassification 是基于 Python 的的图像分类工具，属于 OpenMMLab。
安装 mmcls
请参考 install.md 进行安装。
支持列表
| Model | TorchScript | ONNX Runtime | TensorRT | ncnn | PPLNN | OpenVINO | Model config |
|---|---|---|---|---|---|---|---|
| ResNet | Y | Y | Y | Y | Y | Y | config |
| ResNeXt | Y | Y | Y | Y | Y | Y | config |
| SE-ResNet | Y | Y | Y | Y | Y | Y | config |
| MobileNetV2 | Y | Y | Y | Y | Y | Y | config |
| ShuffleNetV1 | Y | Y | Y | Y | Y | Y | config |
| ShuffleNetV2 | Y | Y | Y | Y | Y | Y | config |
| VisionTransformer | Y | Y | Y | Y | ? | Y | config |
| SwinTransformer | Y | Y | Y | N | ? | N | config |