深度学习训练软件(基于开源开发)分享和使用讲解

深度学习训练软件(基于开源开发)分享和使用讲解

作为一名软件工程师，我经常接触深度学习知识，但由于自己缺乏C++技能，一直没有接触过一些开源项目。

下面我来分享一下我的基于开源代码的深度学习软件。 目前该软件主要支持深度学习的分类功能。 分类函数在我们的实际项目中使用得比较频繁，因为它使用起来比较简单，也很容易上手。 培训软件已上传至云盘。 关注我的公众号，发送“深度学习软件”即可获取下载链接。

下面讲解如何通过软件训练自己的项目。

下载并解压软件并复制文件夹。 该文件夹是项目文件夹，相关的配置参数都在里面。

项目文件夹结构含义如下：

打开项目中的Img->train文件夹，将需要分类的图片复制到对应的文件夹中。 这里我们只对两个标签进行分类（NG、OK）。 当然，也可以对多个标签进行分类。

3、修改Img->.txt文件。 将训练文件中每个文件夹的名称添加到 .txt 中。 如下：

4、以上完成了培训的前期工作。 接下来双击运行可执行文件.exe。

打开界面如下：

5. 打开刚刚配置的项目文件，按照以下步骤训练项目。

图片大小可以使用默认大小。 是否使用GPU取决于您的计算机是否有GPU。 当然，使用GPU可以加快训练速度。 如果使用CPU，速度是相当可观的。

当然你也可以在训练中途停止训练。

如果停止，训练也可以根据之前的训练结果进行继承。 下次训练时软件会提示是否需要继续训练。

培训中遇到的问题及解决方案：

问题1.训练一段时间后，训练停止了。 表现是训练按钮在一段时间后从灰色禁用变为不禁用。 日志显示停止。

这种情况是因为显卡显存无法满足训练需要。 打开项目文件下的配置文件.ini，进行如下修改：

问题2.训练后测试结果不理想。

这种情况是由于训练学习率过高造成的。 打开项目文件下的配置文件.ini，进行如下修改：

6、训练完成后模型加载及效果测试。

按照以下步骤加载项目中训练好的模型（训练时会保存每个阶段的模型，一般我们加载最后一个阶段的模型）

加载所需参数

7. 加载图像。 如果是单张图片预测，直接点击单张预测按钮，选择图片即可。

批量测试时，需要先选择对应的镜像文件路径。 然后点击上一个和下一个按钮即可实现。