AI缺陷检测深度学习训练软件

AIDetection深度学习训练软件

下载链接 链接：百度网盘 请输入提取码 提取码：aidt

YiMing 易明

一、软件分区

图1 AIDetection功能分区

AIDetection主界面分为六大区域，分别为菜单栏、基础功能区、模型训练、训练可视化、数据列表、系统状态。

1） 菜单栏：

菜单栏中包括基础功能区的部分功能，用于方便打开项目和对其他基础功能进行操作。

2） 基础功能区

基础功能区提供项目工程的新建、打开、保存等功能以及数据标注功能和快捷置信度修改。

3） 模型训练区

模型训练区用于对标注好的数据进行管理、模型训练、模型加载、卸载、导出。

4） 训练可视化

训练可视化用于观察模型训练的情况，判断模型训练效果，以便及时调整参数。

5） 数据列表

数据列表用于操作数据和查看标签的详细信息。

6） 系统状态

系统状态用于显示计算机的资源信息。

二、快速开始一个项目

1、新建项目

点击“新建项目”按钮可以快速新建一个工程项目。

图2新建项目

选择想要创建的项目类型（目前只有目标检测，其他暂未开通），填写项目 名称和项目保存的路径以及训练的数据集路径。如需备份数据集到项目路径下，则勾选备份数据集。点击“确定”按钮新建项目成功。

2、数据标注

点击创建矩形按钮进行数据标注。框选缺陷的位置，并输入缺陷种类名称，点击完成按钮标注成功。

图3数据标注

重复以上步骤，将所有需要检测的缺陷类型标记完成。模型的准确率和标注十分相关，尽量将形态相近的缺陷归为一类，形态不同的缺陷要用不同的名称，并绘制缺陷的最小外接矩形。

3、数据集的增删和拆分

如需要增加或替换数据集，点击“数据管理”按钮进行数据集的扩充和替换以及拆分。拆分可将数据集拆成训练和验证两部分，验证部分不参与训练。训练过程中，模型会自动在验证集上评估，并保存评估结果最优的模型。一般将数据拆分为百分之80训练，百分之20验证即可。

图4数据管理

拆分后，列表中红色代表被分为验证集的样本，绿色代表参与训练的样本。

图5数据拆分

4、参数调整

点击“参数调整”按钮进行模型选择。

图6参数调整

在简易设置中可选择效果优先和速度优先两种模式。高级设置中可对模型参数进行专业的调整。如果不使用高级设置，建议打开自适应分辨率和数据增强开关。填写训练轮数，一般数据越多模型越小，需要的轮数越多。训练模式可选择从头训练或继续训练，从头训练会覆盖掉已有模型，继续训练会接着已有模型开始训练。点击“完成”按钮保存。

5、开始训练

点击“开始训练”按钮开始模型训练。在右侧训练曲线和信息栏中可观察模型训练效果和状态。

图7获取许可证文件

如没有许可证文件，点击“是”按钮后，将机器码粘贴到记事本中保存，留言或私信我免费获取许可证。获取许可证后将许可证文件复制到软件所在目录即可。

图8许可证文件存放位置

获取到许可证后，点击“停止训练”“开始训练”即可正常开始模型训练。

6、加载、卸载模型

在模型训练过程中，当有模型保存后可以通过点击“加载模型”按钮查看模型检测效果，切换不同图片可查看效果和检测时间。点击“卸载模型”可以将加载的模型卸载掉。

图9加载模型

7、模型导出

在模型训练好后，点击“模型导出”按钮可将模型打包导出，用于推理库调用。

图10导出模型

如果现场设备是CPU，则导出设备选择CPU，如果是显卡，则导出到对应显卡。这里实际应用的显卡应该和导出到的显卡设备型号相同。选择导出路径，填写好模型备注后点击“确定”按钮导出模型。

8、调用SDK

加入微信交流群。