[发明专利]图片扩增与批注方法、装置及计算机存储介质

说明书

本发明提出一种图片扩增与批注方法，具体包括如下步骤：建立原始图片集；其中，原始图片集内存储有待扩增的原始图片；翻转旋转所述原始图片，得到扩增图片并组成扩增图片集；批注原始图片；按照预设的转换规则对批注后的原始图片进行批注转换得到批注后的扩增图片；合并批注后的原始图片和扩增图片至训练图片集。本发明还对应提供了一种图片扩增与批注装置和计算机存储介质，利用本发明的技术方案，先扩增图片数量再统一图片批注格式，将统一批注后的原始图片和扩增图片用于训练AI深度学习模型，可以简单有效地提高AI深度学习模型的准确度。

技术领域

本发明涉及数据处理技术领域，特别是一种图片扩增与批注方法、装置及计算机存储介质。

背景技术

随着现代社会的发展，使得计算机技术以及相应的软件产品一直处于增长的状态。各行各业在生产过程中均需要使用软件产品，面对庞大的软件产品需求。例如，从产线取得的AI模型训练的图片受限于设备、成本、时间压力等因素使得用于训练的图片数量可能会不够多，进而降低AI深度学习模型的准确度，用于AI深度学习模型的训练图片需统一批注格式。

发明内容

鉴于上述状况，有必要提供一种图片扩增与批注方法、装置及计算机存储介质、装置及计算机存储介质，通过扩增图片数量和统一图片批注格式以提高AI深度学习模型的准确度，以解决上述问题。

本发明第一方面提供了一种图片扩增与批注方法，用于扩增图片数量与批注图片，所述图片扩增与批注包括如下步骤：建立原始图片集；其中，所述原始图片集内存储有待扩增的原始图片；旋转所述原始图片，得到扩增图片并组成扩增图片集；批注所述原始图片；按照预设的转换规则对批注后的所述原始图片进行批注转换得到批注后的所述扩增图片；合并批注后的所述原始图片和所述扩增图片至训练图片集。

优选地，步骤所述旋转所述原始图片，包括如下步骤：翻转所述原始图片得到翻转图片，使所述原始图片与所述翻转图片形成镜像对称；以预设角度分别旋转所述原始图片和所述翻转图片对应得到第一扩增图片和第二扩增图片；其中，所述预设角度大于0°且小于360°。

优选地，所述原始图片和所述翻转图片分别旋转K次对应得到所述第一扩增图片和所述第二扩增图片，每次旋转角度θ＝(2×360)/N度；其中，K＝(N/2)-1，N为图片扩增数量的倍数，N为偶数。

优选地，所述批注所述原始图片步骤包括批注部位的尺寸坐标、点坐标和格式坐标；其中，所述尺寸坐标为所述批注部位的面积在所述原始图片面积中的水平和垂直方向的占比，所述点坐标为所述批注部位中其中一个点的点坐标，所述格式坐标为所述批注部位中水平方向坐标最小值、垂直方向坐标最小值、批注部位宽度和批注部位高度。

优选地，所述按照预设的转换规则对批注后的所述原始图片进行批注转换得到批注后的所述扩增图片步骤，包括：通过所述原始图片的所述批注部位的尺寸坐标的坐标值的转换得到所述扩增图片的批注部位的尺寸坐标；通过所述原始图片的所述批注部位的点坐标值与尺寸坐标值的转换得到所述扩增图片的批注部位的点坐标；通过所述原始图片的所述批注部位的点坐标值与格式坐标值的转换得到所述扩增图片的批注部位的格式坐标。

本发明第二方面提供了一种图片扩增与批注装置，用于扩增图片数量与批注图片，所述图片扩增与批注装置包括：处理器；以及存储器，所述存储器中存储有多个程序模块，多个所述程序模块由所述处理器运行并执行如下步骤：建立原始图片集；其中，所述原始图片集内存储有待扩增的原始图片；旋转所述原始图片，得到扩增图片并组成扩增图片集；批注所述原始图片；按照预设的转换规则对批注后的所述原始图片进行批注转换得到批注后的所述扩增图片；合并批注后的所述原始图片和所述扩增图片至训练图片集。

优选地，多个所述程序模块由所述处理器运行并还执行如下步骤：翻转所述原始图片得到翻转图片，使所述原始图片与所述翻转图片形成镜像对称；以预设角度分别旋转所述原始图片和所述翻转图片对应得到第一扩增图片和第二扩增图片；其中，所述预设角度大于0°且小于360°。