MMClassification 是基于 Python 的的图像分类工具,属于 OpenMMLab。
请参考 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 |