MindSpore计算框架提供了一个官方版本的预训练模型存储库,或者叫做官方版本的预训练模型中心库,那就是 MindSpore / hub 。



首先我们需要明确概念:

第一个就是 mindspore_hub 是一个Python库,或者说是一个Python包(package),

我们需要在Python中调用mindspore_hub库中的函数,来下载官方发布的预训练模型。


官方软件地址(mindspore_hub库地址):

​https://gitee.com/mindspore/hub​



mindspore_hub 库的安装并不依赖计算框架MindSpore, 我们完全可以把mindspore_hub 单独看做是一个下载工具,只不过mindspore_hub是专门给计算框架MindSpore下载预训练模型的。




====================================================



MindSpore_Hub  的安装

其安装方式有多种,不过这里还是建议进行源码安装,官方给出了具体步骤:

  1. 从Gitee下载源码。
git clone https://gitee.com/mindspore/hub.git
  1. 编译安装MindSpore Hub。
cd hub
python setup.py install



安装截图:

MindSpore 框架的官方预训练模型的加载  ——  MindSpore / hub 的安装_python






特别需要注意的一个问题是,MindSpore_Hub 的安装是不依赖MindSpore的,也就是说没有安装MindSpore的环境下也是可以安装MindSpore_Hub 的,

但是  MindSpore_Hub  的使用是需要依赖 MindSpore 的,换句话说在没有MindSpore的环境下成功安装MindSpore_Hub 后也是无法使用的,

所以要使用  MindSpore_Hub  我们还是需要下载安装MindSpore 的。



===================================================================




验证  MindSpore_Hub  的安装是否成功:


正如前面所说我们如果使用MindSpore_Hub 的话需要安装MindSpore, 这里我们假设已经成功安装了MindSpore 。



验证代码:

Python 中执行:

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



运行结果:

WARNING: 'ControlDepend' is deprecated from version 1.1 and will be removed in a future version, use 'Depend' instead.

[WARNING] ME(3634:139972916449408,MainProcess):2021-07-11-14:05:31.331.520 [mindspore/ops/operations/array_ops.py:2302] WARN_DEPRECATED: The usage of Pack is deprecated. Please use Stack.

Downloading data from url https://gitee.com/mindspore/hub/raw/master/mshub_res/assets/mindspore/ascend/0.7/googlenet_v1_cifar10.md


Download finished!

File size = 0.00 Mb

Checking /data/devil/.mscache/mindspore/ascend/0.7/googlenet_v1_cifar10.md...Passed!

remote: Enumerating objects: 38480, done.

remote: Counting objects: 100% (38480/38480), done.

remote: Compressing objects: 100% (1733/1733), done.

remote: Total 388882 (delta 37341), reused 36990 (delta 36742), pack-reused 350402

Receiving objects: 100% (388882/388882), 472.14 MiB | 1.92 MiB/s, done.

Resolving deltas: 100% (316561/316561), done.

From https://gitee.com/mindspore/mindspore

 * [new branch]      master     -> origin/master

 * [new branch]      r0.1       -> origin/r0.1

 * [new branch]      r0.2       -> origin/r0.2

 * [new branch]      r0.3       -> origin/r0.3

 * [new branch]      r0.5       -> origin/r0.5

 * [new branch]      r0.6       -> origin/r0.6

 * [new branch]      r0.7       -> origin/r0.7

 * [new branch]      r1.0       -> origin/r1.0

 * [new branch]      r1.1       -> origin/r1.1

 * [new branch]      r1.2       -> origin/r1.2

 * [new branch]      r1.3       -> origin/r1.3

 * [new tag]               v0.1.0-alpha -> v0.1.0-alpha

 * [new tag]               v0.2.0-alpha -> v0.2.0-alpha

 * [new tag]               v0.3.0-alpha -> v0.3.0-alpha

 * [new tag]               v0.3.1-alpha -> v0.3.1-alpha

 * [new tag]               v0.5.0-beta  -> v0.5.0-beta

 * [new tag]               v0.6.0-beta  -> v0.6.0-beta

 * [new tag]               v0.7.0-beta  -> v0.7.0-beta

 * [new tag]               v1.0.0       -> v1.0.0

 * [new tag]               v1.0.1       -> v1.0.1

 * [new tag]               v1.1.0       -> v1.1.0

 * [new tag]               v1.1.1       -> v1.1.1

 * [new tag]               v1.2.0       -> v1.2.0

 * [new tag]               v1.2.0-rc1   -> v1.2.0-rc1

 * [new tag]               v1.2.1       -> v1.2.1

From https://gitee.com/mindspore/mindspore

 * branch                  master     -> FETCH_HEAD

Downloading data from url https://gitee.com/mindspore/hub/raw/master/mshub_res/assets/mindspore/ascend/0.7/googlenet_v1_cifar10.md


Download finished!

File size = 0.00 Mb

Checking /data/devil/.mscache/mindspore/ascend/0.7/googlenet_v1_cifar10.md...Passed!

File already exists!


进程已结束,退出代码为 0