使用图像分类模块

本章提供 Mech-DLK 示例工程中的冷凝管数据集（ 点击下载 ），带领用户使用“图像分类”模块训练模型，达到示例工程中区分工件的正反面的效果。

1. 新建工程并添加“图像分类”模块：点击初始页面上的 新建工程 ，选择工程路径并输入工程名以新建一个工程。再点击右上角的 ，选择“图像分类”模块。

   

2. 导入工件正反面图像数据集：解压缩下载的数据集压缩包，点击左上方的 导入 ，选择 文件夹 导入下载好的图像数据集。

   

3. 截取 ROI：点击窗口左侧的 ROI 框选选项从图像中框选整个冷凝管作为感兴趣区域，并点击左上方的 应用 确认使用。 截取 ROI 的目的是减少无关背景信息的干扰。

   

4. 创建标签：根据物体名称或特征创建对应的标签，此处需要区分工件正反面，所以直接以正反命名。

5. 标注图像：图像左侧标注工具栏选择对应的标签进行标注，可以同时选中多张图像进行标注。务必确保标签的一致性，禁止出现将图像反面标注为正面的情况。

   

   小技巧

   “图像分类”模块支持同时选中多张图像进行批量标注。

6. 划分训练集与验证集：软件默认将数据集的 80% 划分为训练集，20% 为验证集。需要确保训练集和验证集中都包含 所有种类 的图像，目的是让算法模块在训练过程中可以学习所有种类图像的特征，并对所有种类的图像进行验证。如果默认划分的数据集不满足这一条件，可以点击 拖动滑块手动调整。

   

7. 训练模型：使用默认参数设定，点击 训练 开始训练模型。

   

8. 验证模型：训练结束后，点击 验证 可以验证并查看模型识别效果。

   

9. 导出模型：点击 导出模型 ，选择存放路径即可导出模型 model.dlkpack 到指定文件夹，用户可以根据需求部署模型。