mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity
mmdeploy/docs/codebases/mmcls.md
Yifan Zhou f442dbbe83
[Docs]: MMCLS benchmark (#300) 
* add doc

* Update benchmark

* Update benchmark

* Restore missing words
2021-12-17 14:12:26 +08:00

1.7 KiB
Raw Blame History

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