使用实例分割模块

本章将为用户提供 Mech-DLK 示例工程中的积木数据集 ( 点击下载 ),带领用户使用“实例分割”模块训练模型,达到示例工程中分割不同种类的积木并输出类别标签的效果。

  1. 新建工程并添加“实例分割”模块:点击初始页面上的 新建工程 ,选择工程路径并输入工程名以新建一个工程。再点击右上角的 icon_create ,选择“实例分割”模块。

    ../../../_images/instance_segmentation1.png
  2. 导入积木图像数据集:解压缩下载的数据集压缩包,点击左上方的 导入 ,选择 文件夹 导入下载好的图像数据集。积木图像数据集包含了四种不同形状、颜色的积木。

    ../../../_images/import_images3.png
  3. 截取 ROI:点击选项 icon_roi 从图像中框选盛放积木的料筐作为感兴趣区域,并点击左上方的 应用 确认使用。截取 ROI 的目的是减少无关背景信息的干扰。

    ../../../_images/roi2.png
  4. 创建标签:点击左侧标签栏的 icon_create ,根据物体名称或特征创建对应的标签。此处需要分割不同种类的积木,所以根据积木的不同形状来命名;用户也可以选择根据积木的不同颜色特征命名。

    ../../../_images/create_labels.png
  5. 标注图像:鼠标左键长按或右键单击图像左侧工具栏 icon_tool ,选择适合的标注工具,标注出图像中所有需要分割积木的外轮廓,并确保标注的标签与积木的形状类别对应。

    ../../../_images/labeling.png

    小技巧

    使用多边形工具进行标注,标注完成后,可以拖拽锚点对标注结果进行调整。

  6. 划分训练集与验证集:软件默认将数据集的 80% 划分为训练集,20% 为验证集,可以点击 icon_slider 拖动滑块调整图像占比。需要确保划分后的训练集和验证集中涵盖所有需要分割的种类,如果不满足这一条件,可以选中图像名单击右键进行调整。

    ../../../_images/move_image2.png
  7. 训练模型:使用默认参数设定,点击 训练 开始训练模型。

    ../../../_images/training_chart3.png
  8. 验证模型:训练结束后,点击 验证 可以验证并查看模型识别效果。

    ../../../_images/result_verification3.png
  9. 导出模型:点击 导出模型 选择存放路径即可导出模型到工程文件夹,用户可以根据需求部署模型。

    ../../../_images/model_files3.png