MMClassification is an open-source image classification toolbox based on PyTorch. It is a part of the OpenMMLab project.
Please refer to install.md for installation.
| 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 |