更新时间:2024-09-06 GMT+08:00
分享

模型仓库

Octopus平台支持上传符合平台规范的模型用于标注或训练任务。对智驾模型进行模型微调后,调优后的模型会自动保存到模型仓库中。

上传模型需包含以下内容:

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

新建模型仓库

  1. 在左侧菜单栏中单击“数据资产 > 模型管理”。
  2. 单击“新建模型仓库”,填写基本信息。

    图1 新建模型仓库
    • 名称:包含中英文、数字、“_”“-”,不得超过64个字符。
    • 描述:简要描述模型,不包含“@#$%^&*< > \”,不得超过256个字符。

  3. 选择样本类型和标注。

    图2 选择样本类型和标注
    • 用途:选择训练或标注。
    • 样本类型:当前支持图片、3D点云。
    • 标注:从下拉列表中选择标注。

  4. 选择镜像。

    图3 模型仓库用途选择训练
    图4 模型仓库用途选择标注
    • 镜像用途:模型仓库用途选择训练时,镜像用途为训练任务。模型仓库用途选择标注时,镜像用途可选择预标注或预评审。
    • 镜像仓库:选择镜像仓库以及版本。
    • 共享级别:可选择个人、团队。
      • 个人:只有创建者自己可见可编辑。
      • 团队:租户内所有用户都可见。

  5. 上传模型文件。

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

    图5 上传模型文件
    表1 模型文件相关参数说明

    文件类型

    上传方式

    填写版本名称

    说明

    打开

    本地文件夹形式上传

    需要用户先准备好模型包,具体要求请参见模型文件说明

    关闭

    无需上传

    空模型。

  6. 单击“初始化”,完成模型仓库的创建。
  7. 查看模型仓库详情。

    单击模型仓库名称,进入该模型仓库的详情页面,分为模型仓库详情和版本管理两部分内容。

    • 模型仓库详情:模型仓库的详细信息,如模型ID、名称、描述、用途、来源、样本类型、创建时间、标注等信息。
      图6 模型仓库详情
    • 版本管理:当镜像仓库用途为训练或标注时,展示该模型所有可用版本的信息,如版本名称、在线编辑、状态、打包状态、创建时间等信息,用户可以在线编辑模型版本、新建模型版本和管理模型版本,详见在线编辑模型新建版本模型版本管理。当镜像仓库用途为智驾模型时,展示该模型所有可用版本的信息,如版本名称、描述、状态、创建时间、更新时间等信息,用户可以管理模型版本,详见模型版本管理
      图7 用途为训练、标注的模型版本管理
      图8 用途为智驾模型的模型版本管理

新建版本

Octopus支持用户对已有的模型进行版本更新,步骤如下:

  1. 在“版本管理”模块,单击“新建版本”或在模型列表,单击操作栏中的“新建版本”。
  2. 输入版本名称。

    图9 版本名称
    • 模型版本名称包含中英文、数字、“_”“-”、“.”,不得超过64个字符。
    • 关联已有算法:可选择关联的算法。

  3. 上传本地模型文件。

    将本地模型文件夹拖入框内或单击选择本地模型文件夹

  4. 单击“上传”,页面提示“模型文件已加入上传队列中,等待上传”。

    上传中刷新或关闭浏览器会导致文件上传异常,请谨慎操作!

模型版本管理

在模型仓库详情页的“版本管理”模块,还可以完成以下操作。

表2 训练或标注模型版本管理相关操作

任务

操作步骤

增量训练

单击操作栏“增量训练”,即可跳转到新建训练任务界面,进行增量训练。

部署模型版本

单击操作栏“部署”,打开部署弹出框,默认选中当前模型版本,部署推理服务。部署任务下发成功后,可以到“训练服务”>“推理服务”,查看部署状态。了解部署推理服务参数信息请查看新建推理服务

说明:

支持部署的模型要求:

  • 模型仓库用途为训练
  • 模型版本状态为创建成功

打包

单击操作栏“打包”,提示“模型版本打包中”,待模型打包成功后,当打包状态变为“打包成功”。

下载

当打包状态为“打包成功”时,单击操作栏“下载”,即可将模型下载到本地。

上传

当状态为“初始化”时,可单击版本名称后操作栏内的“上传文件”,选择本地文件上传。

断点续传

当状态为“上传中断”时,可单击版本名称后操作栏内的“断点续传”,选择本地文件重新上传。

删除

单击版本名称后操作栏内的“删除”,删除该版本。版本删除后无法恢复,请谨慎操作。

表3 智驾模型版本管理相关操作

任务

操作步骤

模型微调

当模型状态为“创建成功”,可单击操作栏“模型微调”,对模型进行再次调优操作。

删除模型版本

单击版本名称后操作栏内的“删除”,删除该版本。版本删除后无法恢复,请谨慎操作。

查询模型仓库

可按照“来源”、“样本类型”、“用途”、“共享级别”和“更新时间”过滤模型,也可通过"仓库名称"在搜索框内输入查询内容 ,单击搜索,进行查询查找目标模型。

在线编辑模型

平台提供模型编辑器,在模型详情页,“版本管理”模块的“模型编辑”栏,单击“模型编辑”,进入该模型的在线编辑页面。如图,界面左侧显示的是该模型包内的所有文件,以目录树的形式展示,支持编程语言的渲染,支持MarkDown文件的实时双屏预览。

图10 在线编辑
  • 新建文件夹:选中文件夹并单击,用户将新建一个该文件夹的子文件夹。选中工程文件并单击,将会新建一个新的文件夹,与用户已有的文件夹同级。
  • 新建文件:单击,或右键文件夹,单击“新建文件”。
  • 修改文件:单击,用户可对文件名称进行修改。
  • 删除文件:单击,用户可删除文件。
    • 文件(夹)名称不能为空,且只能包含数字、英文、中文、点、下划线和中划线。
    • 删除后不可恢复,请谨慎操作。
  • 配置界面:单击,按照喜好配置界面基本属性,查看快捷键说明。
    图11 配置界面
  • 删除当前模型文件:单击“删除”,删除当前页面的模型文件。删除后不可恢复,请谨慎操作。
  • 保存模型文件:单击“保存”,保存当前模型。模型更新完毕请及时保存。

相关文档