[发明专利]用于图像处理的神经网络模型训练方法及装置

说明书

本公开提供了一种用于图像处理的神经网络模型训练方法，可以应用于人工智能技术领域。该方法包括：将待处理源图像进行数据增强，添加相同的透明色与不同样式的背景图，得到增强数据集；使用边缘检测算法提取增强数据集中的每个图像的捕捉对象的边缘位置；使用形态学处理方法计算边缘位置对应的边界区域信息，根据边界区域信息截取增强数据集中的每个图像的捕捉对象信息，形成训练数据集；将训练数据集输入到卷积神经网络模型中进行训练，得到图像分类模型。本公开还提供了一种用于图像处理的神经网络模型训练装置、设备、存储介质和程序产品。

技术领域

本公开涉及人工智能领域，更具体地涉及一种用于图像处理的神经网络模型训练方法、装置、设备及介质。

背景技术

移动互联网的蓬勃发展，使得在便携式设备上为用户提供多种类服务成为可能，图像处理等技术的应用平台从PC端逐渐扩展到移动端。充足的计算资源、时间资源和存储资源，是保证模型性能的关键。因此，部署在移动端的图像处理模型需具有占用空间小、准确率高以及实时性好的特点。

典型的深层卷积网络参数量十分庞大，对计算资源要求很高，并不适用于单个终端设备(如摄像头和手机等)，例如AlexNet中全部模型参数的50％都集中在其第一个全连接层上，VGG的模型参数有90％都是全连接层的参数。伴随着移动互联网的发展，为了在便携式设备上使用神经网络，需要减小模型的计算量和参数量，以MobileNet系列为代表的轻量级神经网络应运而生。

在实现本公开构思的过程中，发明人发现相关技术中至少存在如下问题：目前主流的图像分类模型难以实现移动端的图像处理的业务需求。一方面，诸如ResNet-50的卷积神经网络模型具有较高的准确率，但其往往需要较多的资源；另一方面，MobileNet-v2等轻量级神经网络的出现虽然大大降低了资源消耗，但其准确率却无法满足业务需求。因此，如何权衡模型性能(例如准确率)与资源占用(例如存储、计算或时间资源)亟待解决。

发明内容

有鉴于此，本公开提供了一种用于图像处理的神经网络模型训练方法、装置、设备及介质。

本公开的一个方面提供了一种用于图像处理的神经网络模型训练方法，包括：将待处理源图像进行数据增强，添加相同的透明色与不同样式的背景图，得到增强数据集；使用边缘检测算法提取增强数据集中的每个图像的捕捉对象的边缘位置；使用形态学处理方法计算边缘位置对应的边界区域信息，根据边界区域信息截取增强数据集中的每个图像的捕捉对象信息，形成训练数据集；将训练数据集输入到卷积神经网络模型中进行训练，得到图像分类模型。

根据本公开的实施例，数据增强包括仿射变换、颜色变换、高斯噪声或模糊处理中的至少一种。

根据本公开的实施例，仿射变换包括旋转变换、放缩变换、反射变换或剪切变换中的至少一种，放缩变换包括各向同性放缩变换或非各向同性放缩变换。

根据本公开的实施例，边缘检测算法包括基于训练好的整体嵌套边缘检测网络模型的边缘检测算法。

根据本公开的实施例，形态学处理方法包括腐蚀操作、膨胀操作或开闭运算。

根据本公开的实施例，根据边界区域信息截取增强数据集中的每个图像的捕捉对象信息的步骤之后，还包括：将每个图像的边界区域之外的背景区域设置为白色。

根据本公开的实施例，模型训练应用于云端数据中心，方法还包括：根据预设模型切割点，将图像分类模型切分为第一部分模型和第二部分模型；将第一部分模型和第二部分模型分别部署于移动终端和边缘服务器。

根据本公开的实施例，将第一部分模型和第二部分模型分别部署于移动终端和边缘服务器中的步骤之后，还包括：将第一部分模型和第二部分模型分别进行压缩编码。

根据本公开的实施例，卷积神经网络模型包括轻量级网络模型。