mmdeploy/docs/zh_cn/04-supported-codebases/mmcls.md

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