使用Notebook开发模型

使用Notebook开发模型

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

预置代码说明

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

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

导入数据集

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

1. 单击Notebook左侧页签中的“数据集”。

2. 在操作面板中，单击“导入数据集”，系统在弹出框中显示已创建的数据集及其标注情况，如下所示：

   

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

   

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

训练模型

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

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

用户只需要执行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/

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

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

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