线阵相机图像的校正方法、装置、电子设备和存储介质与流程

本发明涉及图像处理，尤其涉及一种线阵相机图像的校正方法、装置、电子设备和存储介质。

背景技术：

1、随着列车技术的飞速发展，线阵相机在列车安全监控领域的应用愈发广泛。线阵相机能够捕捉列车运行过程中的高速动态图像，为列车的安全运行和故障预防提供了重要的视觉信息。然而，由于列车运行环境复杂多变，使得线阵相机获取的图像往往存在畸变等问题，严重影响了图像的质量。因此，对针对列车的线阵相机图像的校正显得尤为重要。

2、相关技术中，如图1所示，采用sift算法分别对待校正图像和模板图像进行特征检测，得到待校正图像的特征点和模板图像的特征点，将待校正图像的特征点与模板图像的特征点进行匹配，得到匹配的至少一个特征点对，将至少一个特征点对中的错误匹配点删除，基于准确匹配的多组特征点对计算变换矩阵，基于变换矩阵进行坐标映射，实现对待校正图像的校正。

3、但列车在运行过程中容易产生污渍，在基于尺度不变特征转换(scale-invariantfeature transform，sift)算法对包括污渍的待校正图像进行特征检测时，会因污渍导致特征提取失败或者特征提取错误，从而降低了图像校正的准确性。

技术实现思路

1、本发明提供一种线阵相机图像的校正方法、装置、电子设备和存储介质，用以解决现有技术中降低图像校正的准确性的缺陷。

2、本发明提供一种线阵相机图像的校正方法，包括：

3、获取列车的待校正线阵相机图像；

4、将模板图像和所述待校正线阵相机图像输入列车部件检测模型，得到所述列车部件检测模型输出的所述模板图像中至少一个第一预设列车部件的第一检测框，以及所述待校正线阵相机图像中至少一个所述第一预设列车部件的第二检测框；所述第一预设列车部件包括表面积大于预设面积的部件；

5、基于各所述第一检测框的位置和对应的所述第二检测框的位置，对所述待校正线阵相机图像进行校正。

6、根据本发明提供的一种线阵相机图像的校正方法，所述获取列车的待校正线阵相机图像，包括：

7、获取列车运行过程中采集的所述列车的初始待校正线阵相机图像，并获取所述列车的真实尺寸；所述真实尺寸包括所述列车的车高和车长；

8、基于所述车高和所述车长，确定所述列车的长高比；

9、基于所述长高比，对所述初始待校正线阵相机图像进行缩放，得到所述待校正线阵相机图像。

10、根据本发明提供的一种线阵相机图像的校正方法，所述真实尺寸还包括各所述第一预设列车部件的第一尺寸和在列车长度方向上两个第二预设列车部件之间的第二尺寸；

11、所述基于所述长高比，对所述初始待校正线阵相机图像进行缩放，得到所述待校正线阵相机图像，包括：

12、基于所述长高比，对所述初始待校正线阵相机图像进行缩放，得到缩放后的待校正线阵相机图像；

13、基于所述第一尺寸和所述第二尺寸，对所述缩放后的待校正线阵相机图像进行调整，得到所述待校正线阵相机图像。

14、根据本发明提供的一种线阵相机图像的校正方法，所述基于所述第一尺寸和所述第二尺寸，对所述缩放后的待校正线阵相机图像进行调整，得到所述待校正线阵相机图像，包括：确定所述第一尺寸中的长度尺寸或径向尺寸与所述车长的第一比值，以及所述第二尺寸与所述车长的第二比值；

15、确定所述第一尺寸中的高度尺寸或径向尺寸与所述车高的第三比值；

16、分别基于所述第一比值、所述第二比值和所述第三比值对所述缩放后的待校正线阵相机图像进行调整，得到所述待校正线阵相机图像。

17、根据本发明提供的一种线阵相机图像的校正方法，所述方法还包括：

18、获取所述列车处于静止状态时采集的所述列车的初始模板图像；

19、基于所述列车的真实尺寸，对所述初始模板图像进行调整，得到所述模板图像。

20、根据本发明提供的一种线阵相机图像的校正方法，所述基于各所述第一检测框的位置和对应的所述第二检测框的位置，对所述待校正线阵相机图像进行校正，包括：

21、针对各所述第一检测框，基于所述第一检测框的位置和对应的所述第二检测框的位置，确定所述第一检测框和对应的所述第二检测框的特征点对；

22、基于各所述特征点对，确定变换矩阵；

23、基于所述变换矩阵，对所述待校正线阵相机图像进行校正。

24、本发明还提供一种线阵相机图像的校正装置，包括：

25、第一获取单元，用于获取列车的待校正线阵相机图像；

26、检测单元，用于将模板图像和所述待校正线阵相机图像输入列车部件检测模型，得到所述列车部件检测模型输出的所述模板图像中至少一个第一预设列车部件的第一检测框，以及所述待校正线阵相机图像中至少一个所述第一预设列车部件的第二检测框；所述第一预设列车部件包括表面积大于预设面积的部件；

27、校正单元，用于基于各所述第一检测框的位置和对应的所述第二检测框的位置，对所述待校正线阵相机图像进行校正。

28、本发明还提供一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述程序时实现如上述任一种所述线阵相机图像的校正方法。

29、本发明还提供一种非暂态计算机可读存储介质，其上存储有计算机程序，该计算机程序被处理器执行时实现如上述任一种所述线阵相机图像的校正方法。

30、本发明还提供一种计算机程序产品，包括计算机程序，所述计算机程序被处理器执行时实现如上述任一种所述线阵相机图像的校正方法。

31、本发明提供的线阵相机图像的校正方法、装置、电子设备和存储介质，将模板图像和列车的待校正线阵相机图像输入列车部件检测模型，得到列车部件检测模型输出的模板图像中至少一个第一预设列车部件的第一检测框，以及待校正线阵相机图像中至少一个第一预设列车部件的第二检测框，且第一预设列车部件包括表面积大于预设面积的部件，基于各第一检测框的位置和对应的第二检测框的位置对待校正线阵相机图像进行校正。可知，本发明是将表面积大于预设面积的预设列车部件对应的检测框的位置作为特征表征，在预设列车部件上存在污渍时，由于预设列车部件的表面积较大，所以能够提高预设列车部件检测框检测的准确性，进一步提高了图像校正的准确性。

技术特征：

1.一种线阵相机图像的校正方法，其特征在于，包括：

2.根据权利要求1所述的线阵相机图像的校正方法，其特征在于，所述获取列车的待校正线阵相机图像，包括：

3.根据权利要求2所述的线阵相机图像的校正方法，其特征在于，所述真实尺寸还包括各所述第一预设列车部件的第一尺寸和在列车长度方向上两个第二预设列车部件之间的第二尺寸；

4.根据权利要求3所述的线阵相机图像的校正方法，其特征在于，所述基于所述第一尺寸和所述第二尺寸，对所述缩放后的待校正线阵相机图像进行调整，得到所述待校正线阵相机图像，包括：

5.根据权利要求1所述的线阵相机图像的校正方法，其特征在于，所述方法还包括：

6.根据权利要求1-5任一项所述的线阵相机图像的校正方法，其特征在于，所述基于各所述第一检测框的位置和对应的所述第二检测框的位置，对所述待校正线阵相机图像进行校正，包括：

7.一种线阵相机图像的校正装置，其特征在于，包括：

8.一种电子设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现如权利要求1至6任一项所述线阵相机图像的校正方法。

9.一种非暂态计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至6任一项所述线阵相机图像的校正方法。

10.一种计算机程序产品，包括计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至6任一项所述线阵相机图像的校正方法。

技术总结本发明提供一种线阵相机图像的校正方法、装置、电子设备和存储介质，涉及图像处理技术领域，其中方法包括：将模板图像和待校正线阵相机图像输入列车部件检测模型，得到列车部件检测模型输出的模板图像中至少一个第一预设列车部件的第一检测框，以及待校正线阵相机图像中至少一个第一预设列车部件的第二检测框；基于各第一检测框的位置和对应的第二检测框的位置，对待校正线阵相机图像进行校正。本发明是将表面积大于预设面积的预设列车部件对应的检测框的位置作为特征表征，在预设列车部件上存在污渍时，由于预设列车部件的表面积较大，所以能够提高预设列车部件检测框检测的准确性，进一步提高了图像校正的准确性。技术研发人员：胡斌,吴正中,张辉,武涛,张东东,张利宽,王晓东受保护的技术使用者：北京城建智控科技股份有限公司技术研发日：技术公布日：2024/11/4