自定义模型

自定义模型

Octopus平台支持您上传在平台支持的训练框架下的自定义模型，目前Octopus平台支持的AI引擎为：Tensorflow和PyTorch，支持的Python版本为3.6；本地上传模型文件时 ，将文件以压缩包的形式上传，上传的方式与算法文件上传方式一致，详情请参见算法管理。

说明： 上传自定义模型需包含以下内容：

• 模型文件；
• 推理脚本，推理脚本命名为：customer_auto_label.py；
• 推理脚本所需依赖库。

添加自定义模型

1. 单击“训练服务>模型管理>创建模型”。

2. 在基本信息部分，填写相应信息。

   

   • 模型名称：包含中英文、数字、“_”“-”，不得超过64个字符；
   • 模型描述：简要描述模型，不包含“@#$%^&*<>/”，不得超过255个字符。

3. 在添加标注部分，从下拉列表中选择标注。

   

4. 在AI引擎部分，从下拉列表中选择AI引擎和样本类型。

   

   • AI引擎：Tensorflow和PyTorch，支持的Python版本为3.6；
   • 样本类型：图片和3D点云。

5. 在模型文件部分，选择文件上传的方式，填写版本名称。

   

   表 1 模型文件相关参数说明

   文件类型	上传方式	填写版本名称
   本地文件	压缩包形式上传	是
   OBS	从OBS桶	是
   空模型	无需上传	否

   • 本地上传和OBS导入模型文件的方式，需要您先准备好模型包，具体要求请参见模型文件说明；
   • 可将使用算法在线训练生成的模型归档到空模型。

6. 输入模型的版本名称，包含中英文、数字、“_”“-”，不得超过64个字符。

7. 单击“创建”，页面提示“创建成功”。

8. 查看模型详情。

   单击模型名称，进入该模型的详情页面，分为模型详情和版本历史两部分内容。

   • 模型详情：模型的详细信息，如模型名称、模型类型、AI引擎、可用版本数量、标注、样本类型等信息。

     

   • 历史版本：展示该模型所有可用版本的信息，如版本ID、版本名称、版本状态、打包状态和创建时间等信息，您可以删除不需要的版本，支持您打包所需版本。

     

添加版本

Octopus支持您对已有的训练模型进行版本更新，步骤如下：

1. 在需要添加版本的模型后“操作”栏单击“添加版本”；

2. 选择模型文件上传方式、填写版本名称信息。

   

   说明： 本地上传模型文件小于5MB，通过OBS上传的模型文件不能超过500M,推荐从OBS桶上传；

3. 单击“确认”，页面提示“添加版本成功”。

模型下载

Octopus支持您将模型文件压缩为.zip的压缩包后，保存并下载到本地。具体操作步骤可参考如下：

前提条件：模型版本创建成功。

1. 单击“训练服务>模型管理”，选择具体的用户自定义模型名称，查看详情页。
2. 在“历史版本”中，找到想要下载打包的版本后，单击操作栏“打包”，提示“模型版本打包中”。
3. 待模型打包成功后，单击操作栏“下载”，即可将模型下载到本地。

删除自定义模型

删除模型：在需要删除的模型后单击“删除”，提示“删除后无法恢复，请谨慎操作”，单击“确认”。

查询自定义模型

您可根据模型名称或AI引擎进行过滤，在搜索框内输入查询内容 ，单击，进行查询。