Yifan Zhou c1ed41c465
[Docs] Refactor document structure (#310)
* Refactor docs

* Fix path

* Update logo
2021-12-21 13:59:07 +08:00

1.7 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 ONNX Runtime TensorRT NCNN PPLNN OpenVINO model config file(example)
ResNet Y Y Y Y Y $MMCLS_DIR/configs/resnet/resnet18_b32x8_imagenet.py
ResNeXt Y Y Y Y Y $MMCLS_DIR/configs/resnext/resnext50_32x4d_b32x8_imagenet.py
SE-ResNet Y Y Y Y Y $MMCLS_DIR/configs/seresnet/seresnet50_b32x8_imagenet.py
MobileNetV2 Y Y Y Y Y $MMCLS_DIR/configs/mobilenet_v2/mobilenet_v2_b32x8_imagenet.py
ShuffleNetV1 Y Y Y Y Y $MMCLS_DIR/configs/shufflenet_v1/shufflenet_v1_1x_b64x16_linearlr_bn_nowd_imagenet.py
ShuffleNetV2 Y Y Y Y Y $MMCLS_DIR/configs/shufflenet_v2/shufflenet_v2_1x_b64x16_linearlr_bn_nowd_imagenet.py

Reminder

None

FAQs

None