81 Star 642 Fork 58

MindSpore / hub

Create your Gitee Account
Explore and code with more than 8 million developers,Free private repositories !:)
Sign up
Clone or Download
README_CN.md 4.53 KB
Copy Edit Web IDE Raw Blame History

MindSpore标志

View English

MindSpore Hub介绍

MindSpore Hub是MindSpore生态的预训练模型应用工具,作为模型开发者和应用开发者的管道:

  • 向模型开发者提供方便快捷的模型发布、提交通道;
  • 向应用开发者提供高质量的预训练模型,结合模型加载以及模型Fine-tune API快速完成模型的迁移到部署的工作。

当前MindSpore Hub提供的预训练模型主要包括 图像分类、目标检测、语义模型、推荐模型等。更多模型内容可以查看官网

特性

  • 即插即用的模型加载: 通过在MindSpore Hub网站上搜索可以快速得到想要的模型,快速体验MindSpore预训练模型;
  • 简单易用的迁移学习: 借助MindSpore灵活的接口,一步实现迁移学习。

环境要求

系统要求和软件依赖

版本号 操作系统 可执行文件安装依赖 源码编译安装依赖
MindSpore Hub master - Ubuntu 18.04 x86_64
- Ubuntu 18.04 aarch64
- EulerOS 2.8 aarch64
- EulerOS 2.5 x86_64
- Python 3.7.5
- MindSpore master
- 其他依赖项参见setup.py
与可执行文件安装依赖相同
  • 在联网状态下,安装whl包时会自动下载setup.py中的依赖项,其余情况需自行安装。

安装

源码安装

  1. 从Gitee下载源码。

    git clone https://gitee.com/mindspore/hub.git
  2. 编译安装MindSpore Hub。

    cd hub
    python setup.py install

二进制文件

使用pip命令安装,hub依赖当前环境中安装的MindSpore版本。

  1. 请从MindSpore Hub下载页面下载并安装whl包。

    pip install mindspore_hub-{version}-py3-none-any.whl
  2. 在能联网的环境中执行以下命令,验证安装结果。

    import mindspore_hub as mshub
    model = mshub.load("mindspore/ascend/0.7/googlenet_v1_cifar10", num_classes=10)

快速入门

参考从Hub加载模型实现模型加载以及模型微调迁移。

文档

有关安装指南、教程和API的更多详细信息,请参阅从Hub加载模型发布模型

社区

MindSpore Hub是MindSpore社区的一部分,关于社区的交流、贡献与治理内容与MindSpore社区一致。

治理

查看MindSpore如何进行开放治理

交流

贡献

欢迎参与贡献。更多详情,请参阅我们的贡献者Wiki

版本说明

版本说明请参阅RELEASE

许可证

Apache License 2.0

FAQ

  • 遇到SSL: CERTIFICATE_VERIFY_FAILED怎么办? 由于你的网络环境,例如你使用代理连接互联网,往往会由于证书配置问题导致python出现ssl verification failed的问题,此时有两种解决方法:
    • 配置好SSL证书**(推荐)**
    • 在加载mindspore_hub前增加如下代码进行折中(最快解决)
import ssl
ssl._create_default_https_context = ssl._create_unverified_context

import mindspore_hub as mshub
model = mshub.load("mindspore/ascend/0.7/googlenet_v1_cifar10", num_classes=10)
1
https://gitee.com/mindspore/hub.git
git@gitee.com:mindspore/hub.git
mindspore
hub
hub
master

Search