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 的安裝
其安裝方式有多種,不過這里還是建議進行源碼安裝,官方給出了具體步驟:
-
從Gitee下載源碼。
git clone https://gitee.com/mindspore/hub.git
-
編譯安裝MindSpore Hub。
cd hub python setup.py install
安裝截圖:
特別需要注意的一個問題是,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