View English

• MindSpore Hub介绍
  • 特性
• 环境要求
  • 系统要求和软件依赖
• 安装
  • 源码安装
  • pip安装
  • 验证是否安装成功
• 快速入门
• 文档
• 社区
  • 治理
  • 交流
• 贡献
• 版本说明
• 许可证
• FAQ

MindSpore Hub介绍

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

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

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

特性

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

环境要求

系统要求和软件依赖

• 在联网状态下，安装whl包时会自动下载setup.py中的依赖项，其余情况需自行安装。

安装

由于Hub直接使用MindSpore models仓代码，Hub对Mindspore版本有依赖。请按照根据下表中所指示的对应关系，在MindSpore下载页面下载并安装对应的whl包。

pip install https://ms-release.obs.cn-north-4.myhuaweicloud.com/{MindSpore-Version}/MindSpore/cpu/ubuntu_x86/mindspore-{MindSpore-Version}-cp37-cp37m-linux_x86_64.whl

源码安装

1. 从Gitee下载源码。

   git clone https://gitee.com/mindspore/hub.git -b r1.9
   

2. 编译安装MindSpore Hub。

   cd hub
   python setup.py install
   

pip安装

使用pip命令安装，请从MindSpore Hub下载页面下载并安装whl包。

pip install mindspore_hub-{version}-py3-none-any.whl

验证是否安装成功

执行以下命令，验证安装结果。导入mindspore_hub模块不报错即安装成功。

import mindspore_hub as mshub

快速入门

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

文档

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

社区

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

治理

查看MindSpore如何进行开放治理。

交流

• MindSpore Slack 开发者交流平台。
• #mindspore IRC频道（仅用于会议记录）
• 视频会议：待定
• 邮件列表：https://mailweb.mindspore.cn/postorius/lists

贡献

欢迎参与贡献。更多详情，请参阅我们的贡献者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/1.6/googlenet_cifar10", num_classes=10)

• 遇到No module named src.*怎么办？ 同一进程中使用load接口加载不同的模型，由于每次加载模型需要将模型文件目录插入到环境变量中，经测试发现：Python只会去最开始插入的目录下查找src.*，尽管你将最开始插入的目录删除，Python还是会去这个目录下查找。解决办法如下：
  • 不添加环境变量，将模型目录下的所有文件都复制到当前工作目录下。

# mindspore_hub_install_path/load.py
def _copy_all_file_to_target_path(path, target_path):
    if not os.path.exists(target_path):
        os.makedirs(target_path)
    path = os.path.realpath(path)
    target_path = os.path.realpath(target_path)
    for p in os.listdir(path):
        copy_path = os.path.join(path, p)
        target_dir = os.path.join(target_path, p)
        _delete_if_exist(target_dir)
        if os.path.isdir(copy_path):
            _copy_all_file_to_target_path(copy_path, target_dir)
        else:
            shutil.copy(copy_path, target_dir)

def _get_network_from_cache(name, path, *args, **kwargs):
    _copy_all_file_to_target_path(path, os.getcwd())
    config_path = os.path.join(os.getcwd(), HUB_CONFIG_FILE)
    if not os.path.exists(config_path):
        raise ValueError('{} not exists.'.format(config_path))
    ......

注意：在load后一个模型时可能会将前一个模型的一些文件替换掉，但是模型训练需保证必要模型文件存在，你必须在加载新模型之前完成对前一个模型的训练。