[发明专利]基于BP神经网络的图像几何畸变的快速识别与校正方法

摘要

本发明揭示了一种基于BP神经网络的图像几何畸变的快速识别与校正方法，通过训练BP神经网络学习得到畸变图像与原图像之间的映射关系，以权值的形式保存在神经网络中，然后构建与原图大小相同的恢复图像，并把恢复图像中的坐标点作为神经网络的输入，得到畸变后的输出，并在畸变图像中利用双线性插值法确定像素值。本发明提出的基于BP神经网络的图像几何畸变的快速识别与校正方法，利用BP神经网络可以直接从畸变图像中学习得到确定的畸变方式，要比用畸变模板得到的准确得多。

主权项

一种基于BP神经网络的图像几何畸变的快速识别与校正方法，其特征在于，通过训练BP神经网络学习得到畸变图像与原图像之间的映射关系，以权值的形式保存在神经网络中，然后构建与原图大小相同的恢复图像，并把恢复图像中的坐标点作为神经网络的输入，得到畸变后的输出，并在畸变图像中利用双线性插值法确定像素值；所述方法具体包括如下步骤：S1、搭建三层BP神经网络，其中输入层有2个神经元，隐藏层有45个神经元，输出层有2个神经元；S2、输入层与隐藏层之间利用sigmoid函数作为激活函数，即f(y)＝1/(1+exp(‑1*x))；其中，f(y)是隐藏层的输出，x是输入与权值之间的向量积；隐藏层与输出层之间利用pureline函数作为激活函数，即隐藏层输出与隐藏层与输出层之间的向量积既是最终的输出；S3、分别在未畸变图像a和畸变图像b中选取相对应的控制点作为神经网络的训练样本，控制点的选取是通过matlab的内部函数cpselect进行逐点选取的；S4、利用训练样本训练神经网络，当迭代次数达到预设值或者误差值小于给定阈值时，停止训练，并把此时神经网络的权值保存到txt文本中；S5、构造一个与原图像大小相同的恢复图像，把恢复图像中任一点的坐标(x，y)作为神经网络的输入，得到的输出点(xd，yd)在畸变图像中利用双线性插值法确定其像素值，也即为恢复图像中该点的像素值；S6、对恢复图像中的每一个点都进行步骤S5的操作。