查询模型
使用约束
- 支持的模型信息是从昇腾开发者社区的ModelZoo获取。
MindStudio后台是通过ModelZoo.ini文件中的URL:https://apigw-01.huawei.com/api/ascendcommunity/modelzoo/quickCheck使用包含X-HW-ID:com.huawei.biz.idp_public的请求头来获取昇腾开发者社区ModelZoo上支持的模型信息。
- 如果当前未安装有效的Ascend-cann-toolkit开发套件包,将不显示步骤1中两种入口,请检查Ascend-cann-toolkit开发套件包是否正确安装。
模型查询
- 进入模型查询界面,两种进入方式如下所示。
- 在菜单栏选择 。
- 在工具栏单击图标。
- 如果用户环境可以联网,成功获取后台信息,则弹出图1所示。界面参数信息说明如表1所示。 界面,显示预训练好的模型,如
如果安装的CANN为6.0.RC1之前的版本,且界面显示中文乱码,请参见中文显示乱码和界面显示不全不规整处理。
表1 参数说明 参数
说明
Model Name
模型名称。
Application Area
应用领域,例如Image Classification、Object Detection。
Categories Id
应用级别,例如Official。
Update Time
更新时间。
Precision
模型精度,例如FP16、INT8。
Model Type
模型类型,例如Offline Model。
Frame
框架,例如PyTorch、TensorFlow。
Processor Type
处理器类型,例如Atlas 200/300/500 推理产品、Ascend 910。
Version
版本。
Download Link
下载链接,可以复制对应的链接粘贴到浏览器上打开,获取相应的模型文件。
Model Format
模型格式,例如om、onnx。
File Size
文件大小。
可以通过“Search by”后的下拉框选择支持的信息字段并在输入框中补充关键字进行模型信息的搜索和筛选。
获取后台信息失败或者获取ModelZoo的配置文件URL被修改时,请参考如下方法处理。
父主题: 算子和模型速查