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))
......
View English
MindSpore Hub介绍
MindSpore Hub是MindSpore生态的预训练模型应用工具,作为模型开发者和应用开发者的管道:
当前MindSpore Hub提供的预训练模型主要包括 图像分类、目标检测、语义模型、推荐模型等。更多模型内容可以查看官网。
特性
环境要求
系统要求和软件依赖
- Ubuntu 18.04 aarch64
- EulerOS 2.8 aarch64
- EulerOS 2.5 x86_64
- MindSpore master
- 其他依赖项参见setup.py
setup.py中的依赖项,其余情况需自行安装。安装
由于Hub直接使用MindSpore models仓代码,Hub对Mindspore版本有依赖。请按照根据下表中所指示的对应关系,在MindSpore下载页面下载并安装对应的whl包。
源码安装
从Gitee下载源码。
编译安装MindSpore Hub。
pip安装
使用
pip命令安装,请从MindSpore Hub下载页面下载并安装whl包。验证是否安装成功
执行以下命令,验证安装结果。导入mindspore_hub模块不报错即安装成功。
快速入门
参考从Hub加载模型实现模型加载以及模型微调迁移。
文档
有关安装指南、教程和API的更多详细信息,请参阅从Hub加载模型和发布模型。
社区
MindSpore Hub是MindSpore社区的一部分,关于社区的交流、贡献与治理内容与MindSpore社区一致。
治理
查看MindSpore如何进行开放治理。
交流
#mindsporeIRC频道(仅用于会议记录)贡献
欢迎参与贡献。更多详情,请参阅我们的贡献者Wiki。
版本说明
版本说明请参阅RELEASE。
许可证
Apache License 2.0
FAQ
SSL: CERTIFICATE_VERIFY_FAILED怎么办? 由于你的网络环境,例如你使用代理连接互联网,往往会由于证书配置问题导致python出现ssl verification failed的问题,此时有两种解决方法:No module named src.*怎么办? 同一进程中使用load接口加载不同的模型,由于每次加载模型需要将模型文件目录插入到环境变量中,经测试发现:Python只会去最开始插入的目录下查找src.*,尽管你将最开始插入的目录删除,Python还是会去这个目录下查找。解决办法:不添加环境变量,将模型目录下的所有文件都复制到当前工作目录下。注意:在load后一个模型时可能会将前一个模型的一些文件替换掉,但是模型训练需保证必要模型文件存在,你必须在加载新模型之前完成对前一个模型的训练。