AllentDan 0111e1f10f
[Feature] add swin for cls (#911)
* add swin for cls

* add ut and doc

* reduce trt batch size

* add regression test

* resolve comments

* remove useless rewriting logic
2022-08-23 17:14:24 +08:00

2.3 KiB

MMClassification Support

MMClassification is an open-source image classification toolbox based on PyTorch. It is a part of the OpenMMLab project.

MMClassification installation tutorial

Please refer to install.md for installation.

List of MMClassification models supported by MMDeploy

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