神经网络训练方法、设备及计算机可读介质与流程

技术特征：
1.一种神经网络训练方法，其特征在于，该方法包括：将训练图片进行分形处理，获取包括四个旋转角度的所述训练图片的输入图片，所述四个旋转角度分别为0
°
、90
°
、180
°
和270
°
；将所述输入图片输入所述神经网络进行前向传播，提取所述输入图片对应的输出特征向量，所述输出特征向量包括第一特征向量、第二特征向量、第三特征向量和第四特征向量，所述第一特征向量表示旋转角度为0
°
的训练图片的图像特征，所述第二特征向量表示旋转角度为90
°
的训练图片的图像特征，所述第三特征向量表示旋转角度为180
°
的训练图片的图像特征，所述第四特征向量表示旋转角度为270
°
的训练图片的图像特征；根据所述输出特征向量计算对应于不同旋转角度的损失函数值，并根据所述损失函数值计算梯度，进行反向传播以更新所述神经网络的参数；重复进行所述前向传播和反向传播，直至所述神经网络收敛。2.根据权利要求1所述的方法，其特征在于，将训练图片进行分形处理，获取包括四个旋转角度的所述训练图片的输入图片，所述四个旋转角度分别为0
°
、90
°
、180
°
和270
°
，包括：将同一训练图片分别旋转90
°
、180
°
和270
°
，以获得旋转角度为90
°
、180
°
和270
°
的训练图片；将未进行旋转的训练图片与所述旋转角度为90
°
、180
°
和270
°
的训练图片进行组合，获得包括四个旋转角度的所述训练图片的输入图片，所述四个旋转角度分别为0
°
、90
°
、180
°
和270
°
。3.根据权利要求2所述的方法，其特征在于，将未进行旋转的训练图片与所述旋转角度为90
°
、180
°
和270
°
的训练图片进行组合，获得包括四个旋转角度的所述训练图片的输入图片，所述四个旋转角度分别为0
°
、90
°
、180
°
和270
°
，包括：将未进行旋转的训练图片与所述旋转角度为90
°
、180
°
和270
°
的训练图片，按照[[p1,p2],[p4,p3]]的顺序组合为图片矩阵，并将所述图片矩阵作为包括四个旋转角度的所述训练图片的输入图片，其中，p1、p2、p3和p4分别是旋转角度为0
°
、90
°
、180
°
和270
°
的训练图片。4.根据权利要求1所述的方法，其特征在于，所述神经网络的结构包括骨干网络、全连接层和标准化层；将所述输入图片输入所述神经网络进行前向传播，提取所述输入图片对应的输出特征向量，包括：将所述输入图片输入骨干网络，输出n维的特征向量；将n维的特征向量输入全连接层和标准化层，输出四组m维的特征向量，其中，每组m维的特征向量分别为输入特征向量中的第一特征向量、第二特征向量、第三特征向量和第四特征向量。5.根据权利要求4所述的方法，其特征在于，所述全连接层分别包括4个全连接单元，所述标准化层包括4个标准化单元；将n维的特征向量输入全连接层和标准化层，输出4组m维的特征向量，其中，每组m维的特征向量分别为输入特征向量中的第一特征向量、第二特征向量、第三特征向量和第四特征向量，包括：
将n维的特征向量输入全连接层中的各个全连接单元输出四组m维的特征向量，并将每组m维的特征向量输入标准化层中对应的一个标准化单元中，分别输出第一特征向量、第二特征向量、第三特征向量和第四特征向量。6.根据权利要求4所述的方法，其特征在于，所述骨干网络采用bn
‑
inception网络。7.根据权利要求1所述的方法，其特征在于，根据所述输出特征向量计算对应于不同旋转角度的损失函数值，并根据所述损失函数值计算梯度，进行反向传播以更新所述神经网络的参数，包括：分别根据第一特征向量、第二特征向量、第三特征向量和第四特征向量计算第一损失函数值、第二损失函数值、第三损失函数值和第四损失函数值；计算所述第一损失函数值、第二损失函数值、第三损失函数值和第四损失函数值的平均值，并根据所述平均值计算梯度，进行反向传播以更新所述神经网络的参数。8.根据权利要求7所述的方法，其特征在于，计算所述损失函数值的损失函数为ms
‑
loss函数。9.一种计算机设备，其特征在于，该设备包括用于存储计算机程序指令的存储器和用于执行程序指令的处理器，其中，当该计算机程序指令被该处理器执行时，使该设备执行权利要求1至8中任一项所述的方法的步骤。10.一种计算机可读介质，其特征在于，其上存储有计算机可读指令，所述计算机可读指令可被处理器执行以实现如权利要求1至8中任一项所述的方法的步骤。

技术总结
本申请提供了一种神经网络训练方案，该方案会将训练图片先进行分形处理，获取包括多旋转角度的所述训练图片的输入图片，使得提取到的图像特征除了包含图片的内容信息之外，还包含了方向性信息，由此训练所获得的神经网络可以识别出更加立体的图像特征，即包括内容信息和方向性信息，从而可以提升神经网络的识别性能，使其可以更好的识别具有一定旋转角度的图片。片。片。


技术研发人员：姚广 苏仲岳 闫正
受保护的技术使用者：上海眼控科技股份有限公司
技术研发日：2021.07.15
技术公布日：2021/10/23