开放能力
开发平台
行业应用
生态合作
开发与教学
资讯 社区 控制台
技术能力
语音技术
图像技术
文字识别
人脸与人体识别
视频技术
AR与VR
自然语言处理
知识图谱
数据智能
场景方案
部署方案
行业应用
智能教育
智能医疗
智能零售
智能工业
企业服务
智能政务
智能农业
信息服务
智能园区
智能硬件
BML 全功能AI开发平台

    使用Notebook开发模型

    使用Notebook开发模型

    在Notebook可以导入已创建的数据集并将其用于模型训练。

    预置代码说明

    当您创建的是图像分类、物体检测类型的Notbeook时,系统会自动生成一份代码,您只需要导入数据集并将代码做简单的适配即可快速创建。

    当您创建的是通用型的Notebook时,系统会根据您使用的框架来生成一份示例代码。

    导入数据集

    通过导入操作可以将已创建的数据集下载到Notebook的运行环境中,导入的数据集会保存在“data”目录下以数据集ID命名的文件夹中,对于已标注的图片,下载内容包括原始图片文件和同名的json标注文件。

    1. 单击Notebook左侧页签中的“数据集”。
    2. 在操作面板中,单击“导入数据集”,系统在弹出框中显示已创建的数据集及其标注情况,如下所示:

      fig05.png

    3. 勾选待导入数据集前的复选框并单击“确认”,开始导入数据集,如下所示:

      fig06.png

      数据集导入操作是单向的,且当数据服务中的数据集有更新时,不会自动同步,需要手动重新刷新。

    训练模型

    用户可以在Notebook或终端中编写模型训练代码,并完成模型的训练。

    本节给出如何使用示例代码完成导入数据集的建模。示例代码结构如下所示:

    fig002.png

    用户只需要执行train.sh脚本即可启动模型训练,其输入参数如下所示:

    train.sh <数据目录> <模型保存目录>

    • 数据目录:数据集保存目录的绝对路径
    • 模型保存目录:请设置为/home/work/PretrainedModel/,否则会造成生成新模型版本时无法读取模型文件。

    上述代码结构不是必须的,用户可以自行编写代码训练模型,只需要将模型保存到/home/work/PretrainedModel/目录即可。 使用示例代码完成模型训练的步骤如下所示:

    使用示例代码完成模型训练的步骤如下所示:

    1. 在Notebook中执行如下代码查看已导入的数据集。

      ! ls /home/work/data/

    2. 在Notebook中执行train.sh脚本进行模型训练。

      ! bash demo/train.sh /home/work/data/96159/ /home/work/PretrainedModel/

    请根据实际情况修改上述代码中的数据集路径。

    1. 查看生成的模型文件。

      在Notebook左侧导航栏中单击“文件夹”,并打开“Pretrainedmodel”文件夹,如下所示:

      fig003.png

    上一篇
    创建并启动Notebook
    下一篇
    保存Notebook中的模型