# models

**Repository Path**: osbo/models

## Basic Information

- **Project Name**: models
- **Description**: Models of MindSpore
- **Primary Language**: Unknown
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 1448
- **Created**: 2021-09-18
- **Last Updated**: 2021-11-03

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# ![MindSpore Logo](https://www.mindspore.cn/static/img/logo_black.6a5c850d.png)

## Welcome to the Model Zoo for MindSpore

In order to facilitate developers to enjoy the benefits of MindSpore framework, we will continue to add typical networks and some of the related pre-trained models. If you have needs for the model zoo, you can file an issue on [gitee](https://gitee.com/mindspore/mindspore/issues) or [MindSpore](https://bbs.huaweicloud.com/forum/forum-1076-1.html), We will consider it in time.

- SOTA models using the latest MindSpore APIs

- The  best benefits from MindSpore

- Officially maintained and supported

## Table of Contents

### Official

|  Domain | Sub Domain    | Network  | Ascend  | GPU | CPU |
|:------   |:------| :-----------  |:------:   |:------:  |:-----: |
|Computer Vision (CV) | Image Classification  | [AlexNet](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/alexnet)          |  ✅ |  ✅ |   |
| Computer Vision (CV)  | Image Classification  | [CNN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/cnn_direction_model)  |  ✅ |    |   |
| Computer Vision (CV)  | Image Classification  | [DenseNet100](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/densenet) |    |    | ✅ |
| Computer Vision (CV)  | Image Classification  | [DenseNet121](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/densenet) |  ✅ |    |   |
| Computer Vision (CV)  | Image Classification  | [DPN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/dpn) |  ✅ |    |   |
| Computer Vision (CV)  | Image Classification  | [EfficientNet-B0](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/efficientnet) |    |  ✅ |   |
| Computer Vision (CV)  | Image Classification  | [GoogLeNet](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/googlenet)    |  ✅  | ✅ |   |
| Computer Vision (CV)  | Image Classification  | [InceptionV3](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/inceptionv3)   |  ✅ |    |   |
| Computer Vision (CV)  | Image Classification  | [InceptionV4](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/inceptionv4)    |  ✅ |    |   |
| Computer Vision (CV)  | Image Classification  | [LeNet](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/lenet)              |  ✅ |  ✅ | ✅ |
| Computer Vision (CV)  | Image Classification  | [LeNet (Quantization)](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/lenet_quant)      |  ✅ |  ✅ |   |
| Computer Vision (CV)  | Image Classification  | [MobileNetV1](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/mobilenetv1)      |  ✅ |    |   |
| Computer Vision (CV)  | Image Classification  | [MobileNetV2](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/mobilenetv2)      |  ✅ |  ✅ | ✅ |
| Computer Vision (CV)  | Image Classification  | [MobileNetV2 (Quantization)](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/mobilenetv2_quant)   |  ✅ |  ✅ |   |
| Computer Vision (CV)  | Image Classification  | [MobileNetV3](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/mobilenetv3)  |    |  ✅ |   |
| Computer Vision (CV)  | Image Classification  | [NASNet](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/nasnet) |    |  ✅ |   |
| Computer Vision (CV)  | Image Classification  | [ResNet-18](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/resnet)          |  ✅ |    |   |
| Computer Vision (CV)  | Image Classification  | [ResNet-50](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/resnet)          |  ✅ |  ✅ | ✅ |
| Computer Vision (CV)  | Image Classification  | [ResNet-50 (Quantization)](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/resnet50_quant)          |  ✅ |    |   |
|Computer Vision (CV)  | Image Classification  | [ResNet-101](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/resnet)       |  ✅ | ✅ |   |
|Computer Vision (CV)  | Image Classification  | [ResNeXt50](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/resnext)     |  ✅ | ✅ |   |
|Computer Vision (CV)  | Image Classification  | [SE-ResNet50](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/resnet)      |  ✅ |   |   |
| Computer Vision (CV)  | Image Classification  | [ShuffleNetV1](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/shufflenetv1)  |  ✅ |    |   |
| Computer Vision (CV)  | Image Classification  | [ShuffleNetV2](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/shufflenetv2) |    |  ✅ |   |
| Computer Vision (CV)  | Image Classification  | [SqueezeNet](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/squeezenet) |  ✅ |    |   |
| Computer Vision (CV)  | Image Classification  | [Tiny-DarkNet](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/tinydarknet)       |  ✅ |    |   |
| Computer Vision (CV)  | Image Classification  | [VGG16](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/vgg16)                |  ✅ |  ✅ |   |
| Computer Vision (CV)  | Image Classification  | [Xception](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/xception)             |  ✅ |    |   |
| Computer Vision (CV) | Object Detection  | [CenterFace](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/centerface)     |  ✅ |    |   |
| Computer Vision (CV) | Object Detection  | [CTPN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/ctpn)     |  ✅ |    |   |
| Computer Vision (CV)  | Object Detection  | [Faster R-CNN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/faster_rcnn)   |  ✅ |  ✅ |   |
| Computer Vision (CV)  | Object Detection  | [Mask R-CNN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/maskrcnn)         |  ✅ |    |   |
| Computer Vision (CV)  | Object Detection  | [Mask R-CNN (MobileNetV1)](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/maskrcnn_mobilenetv1)    |  ✅ |    |   |
| Computer Vision (CV) | Object Detection  | [RetinaFace-ResNet50](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/retinaface_resnet50)     |    |  ✅ |   |
| Computer Vision (CV)  | Object Detection  | [SSD](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/ssd)                   |  ✅ | ✅ | ✅ |
| Computer Vision (CV)  | Object Detection  | [SSD-MobileNetV1-FPN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/ssd)         |  ✅ |    |   |
| Computer Vision (CV)  | Object Detection  | [SSD-Resnet50-FPN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/ssd)                   |  ✅ |   |   |
| Computer Vision (CV)  | Object Detection  | [SSD-VGG16](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/ssd)                   |  ✅ |   |   |
| Computer Vision (CV) | Object Detection  | [WarpCTC](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/warpctc)                    |  ✅ |  ✅ |   |
| Computer Vision (CV)  | Object Detection  | [YOLOv3-ResNet18](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/yolov3_resnet18)    | ✅ |    |   |
| Computer Vision (CV)  | Object Detection  | [YOLOv3-DarkNet53](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/yolov3_darknet53)         |  ✅ |  ✅ |   |
| Computer Vision (CV)  | Object Detection  | [YOLOv3-DarkNet53 (Quantization)](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/yolov3_darknet53_quant)  |  ✅ |    |   |
| Computer Vision (CV)  | Object Detection  | [YOLOv4](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/yolov4)         |  ✅ |    |   |
| Computer Vision (CV) | Text Detection  | [DeepText](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/deeptext)   |  ✅ |    |   |
| Computer Vision (CV) | Text Detection  | [PSENet](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/psenet)   |  ✅ |    |   |
| Computer Vision (CV) | Text Recognition  | [CNN+CTC](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/cnnctc)                |  ✅ |    |   |
| Computer Vision (CV) | Semantic Segmentation  | [DeepLabV3](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/deeplabv3)     |  ✅ |    | ✅ |
| Computer Vision (CV) | Semantic Segmentation  | [U-Net2D (Medical)](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/unet)                |  ✅ |    |   |
| Computer Vision (CV) | Semantic Segmentation  | [U-Net3D (Medical)](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/unet3d)                |  ✅ |    |   |
| Computer Vision (CV) | Semantic Segmentation  | [U-Net++](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/unet)                |  ✅ |    |   |
| Computer Vision (CV) | Keypoint Detection  | [OpenPose](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/openpose)                |  ✅ |    |   |
| Computer Vision (CV) | Keypoint Detection  | [SimplePoseNet](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/simple_pose)                |  ✅ |    |   |
| Computer Vision (CV) | Optical Character Recognition  | [CRNN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/cv/crnn)                |  ✅ |    |   |
| Natural Language Processing (NLP) | Natural Language Understanding  | [BERT](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/nlp/bert)        |  ✅ |  ✅ |   |
| Natural Language Processing (NLP) | Natural Language Understanding  | [FastText](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/nlp/fasttext)        |  ✅ |    |   |
| Natural Language Processing (NLP) | Natural Language Understanding  | [GNMT v2](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/nlp/gnmt_v2)        |  ✅ |    |   |
| Natural Language Processing (NLP) | Natural Language Understanding  | [GRU](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/nlp/gru)                |  ✅ |    |   |
| Natural Language Processing (NLP) | Natural Language Understanding  | [MASS](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/nlp/mass)   |  ✅ |  ✅ |   |
| Natural Language Processing (NLP) | Natural Language Understanding  | [SentimentNet](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/nlp/lstm)      |  ✅ |  ✅ | ✅ |
| Natural Language Processing (NLP) | Natural Language Understanding  | [Transformer](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/nlp/transformer)   |  ✅ |  ✅ |   |
| Natural Language Processing (NLP) | Natural Language Understanding  | [TinyBERT](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/nlp/tinybert)       |  ✅ |  ✅ |   |
| Natural Language Processing (NLP) | Natural Language Understanding  | [TextCNN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/nlp/textcnn)                |  ✅ |    |   |
| Recommender | Recommender System, CTR prediction  | [DeepFM](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/recommend/deepfm)                               |  ✅ |  ✅ | ✅ |
| Recommender | Recommender System, Search, Ranking  | [Wide&Deep](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/recommend/wide_and_deep)             |  ✅ |  ✅ |   |
| Recommender | Recommender System  | [NAML](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/recommend/naml)             |  ✅ |    |   |
| Recommender | Recommender System  | [NCF](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/recommend/ncf)             |  ✅ |   |   |
| Graph Neural Networks (GNN) | Text Classification  | [GCN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/gnn/gcn)  |  ✅ |    |   |
| Graph Neural Networks (GNN) | Text Classification  | [GAT](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/gnn/gat)  |  ✅ |    |   |
| Graph Neural Networks (GNN) | Recommender System | [BGCF](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/official/gnn/bgcf) |  ✅ |    |   |

### Research

|  Domain | Sub Domain    | Network  | Ascend | GPU | CPU |
|:------   |:------| :-----------   |:------:   |:------:  |:-----:  |
| Computer Vision (CV) | Image Classification  | [FaceAttributes](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/cv/FaceAttribute)     |  ✅ |    |   |
| Computer Vision (CV) | Object Detection  | [FaceDetection](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/cv/FaceDetection)     |  ✅ |    |   |
| Computer Vision (CV) | Image Classification  | [FaceQualityAssessment](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/cv/FaceQualityAssessment)     |  ✅ |    |   |
| Computer Vision (CV) | Image Classification  | [FaceRecognition](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/cv/FaceRecognition)     |  ✅ |    |   |
| Computer Vision (CV) | Image Classification  | [FaceRecognitionForTracking](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/cv/FaceRecognitionForTracking)     |  ✅ |    |   |
| Computer Vision (CV) | Object Detection  | [Spnas](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/cv/Spnas)     |  ✅ |    |   |
| Computer Vision (CV) | Object Detection  | [SSD-GhostNet](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/cv/ssd_ghostnet)               |  ✅ |    |   |
| Computer Vision (CV)  | Key Point Detection  | [CenterNet](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/cv/centernet)               |  ✅ |   | ✅ |
| Computer Vision (CV)  | Image Style Transfer  | [CycleGAN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/cv/CycleGAN)             |        |   | ✅ |
| Natural Language Processing (NLP) | Natural Language Understanding  | [DS-CNN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/nlp/dscnn)      |  ✅ |    |   |
| Natural Language Processing (NLP) | Natural Language Understanding  | [TextRCNN](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/nlp/textrcnn)  |  ✅ |    |   |
| Natural Language Processing (NLP) | Natural Language Understanding  | [TPRR](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/nlp/tprr)  |  ✅ |    |   |
| Recommender | Recommender System, CTR prediction | [AutoDis](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/recommend/autodis)          |  ✅ |    |   |
| Audio | Audio Tagging | [FCN-4](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/audio/fcn-4)   |  ✅ |    |   |
| High Performance Computing | Molecular Dynamics | [DeepPotentialH2O](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/hpc/molecular_dynamics)   |  ✅ |    |   |
| High Performance Computing | Ocean Model | [GOMO](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/research/hpc/ocean_model)   |    |  ✅ |   |

- [Community](https://gitee.com/mindspore/mindspore/tree/master/model_zoo/community)

## Announcements

### 2021.9.15 Set up repository `models`

`models` comes from the directory `model_zoo` of repository [mindspore](https://gitee.com/mindspore/mindspore). This new repository doesn't contain any history of commits about the directory `model_zoo` in `mindspore`, you could refer to the repository `mindspore` for the past commits.

## Related Website

Here is the ModelZoo for MindSpore which support different devices including Ascend, GPU, CPU and mobile.

If you are looking for exclusive models only for Ascend using different ML platform, you could refer to [Ascend ModelZoo](https://hiascend.com/software/modelzoo) and corresponding [gitee repository](https://gitee.com/ascend/modelzoo)

## Disclaimers

Mindspore only provides scripts that downloads and preprocesses public datasets. We do not own these datasets and are not responsible for their quality or maintenance. Please make sure you have permission to use the dataset under the dataset’s license. The models trained on these dataset are for non-commercial research and educational purpose only.

To dataset owners: we will remove or update all public content upon request if you don’t want your dataset included on Mindspore, or wish to update it in any way. Please contact us through a Github/Gitee issue. Your understanding and contribution to this community is greatly appreciated.

MindSpore is Apache 2.0 licensed. Please see the LICENSE file.

## License

[Apache License 2.0](https://gitee.com/mindspore/mindspore/blob/master/LICENSE)

## FAQ

- **Q: How to resolve the lack of memory while using `PYNATIVE_MODE` with errors such as *Failed to alloc memory pool memory*?**

  **A**: `PYNATIVE_MODE` usually requires more memory than `GRAPH_MODE`, especially in training process which have to deal with back propagation. You could try using smaller batch size.

- **Q: How to resolve the error about the interface not supported, such as `cann not import`?**

  **A**: Please check the version of MindSpore and the branch you fetch the modelzoo scripts. Some model scripits in latest branch will use new interface in the latest version of MindSpore.