本发明涉及图像处理,尤其涉及一种图像校正方法、装置、计算机设备及存储介质。
背景技术:
1、计算机视觉在工业检测中的应用越来越广泛。获取待检测图像时,工业相机与待检测物体之间存在偏差,导致待检测图像倾斜。因此需要对待检测图像进行校正,以提高检测的精确度。
2、相关技术中,基于霍夫变换的方法对待检测图像进行图像校正。然而,该方法的精确度有待提升。
技术实现思路
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、在其中一个实施方式,所述根据所述修正特征点集合与所述第二特征点集合,确定是否从所述修正特征点集合中删除所述待剔除特征点,包括:
32、从所述修正特征点集合中尝试删除所述待剔除特征点,得到删除后特征点集合;
33、在所述第二特征点集合中确定与所述删除后特征点集合中各特征点对应的核实用特征点,得到核实特征点集合;
34、若所述删除后特征点集合与所述核实特征点集合间的距离偏差和小于偏差和阈值,确定删除所述待剔除特征点;
35、若所述删除后特征点集合与所述核实特征点集合间的距离偏差和不小于偏差和阈值,确定不删除所述待剔除特征点。
36、本说明书实施方式提供一种图像校正装置,所述装置包括:
37、特征点匹配模块,用于获取待检测图像的第一特征点集合与参考图像的第二特征点集合进行特征点匹配,得到所述第一特征点集合中与所述第二特征点集合匹配的目标特征点;
38、分布情况确定模块,用于若所述目标特征点的数量小于预设数量阈值,确定所述第二特征点集合中的第二特征点在所述参考图像上的分布情况;
39、特征点配对模块,用于根据所述分布情况、所述第一特征点集合以及所述第二特征点集合进行聚类和特征点配对,得到多个特征点匹配对;其中,所述特征点匹配对的数量大于所述预设数量阈值;
40、待检测图像校正模块,用于基于所述多个特征点匹配对所述待检测图像进行校正,得到校正后检测图像。
41、本发明提供一种就计算机设备,所述计算机设备包括:存储器,以及与所述存储器通信连接的一个或多个处理器;所述存储器中存储有可被所述一个或多个处理器执行的指令,所述指令被所述一个或多个处理器执行,以使所述一个或多个处理器实现上述任一项实施方式所述的方法的步骤。
42、本发明提供计算机可读存储介质,其上存储有计算机程序,所述计算机程序被处理器执行时实现上述任一项实施方式所述的方法的步骤。
43、本发明提供一种计算机程序产品,所述计算机程序产品中包括指令,所述指令被计算机设备的处理器执行时,使得所述计算机设备能够执行上述任一项实施方式所述的方法的步骤。
44、本说明书实施方式提供一种芯片,包括存储单元和处理单元,所述存储单元存储有计算机程序,所述处理单元执行所述计算机程序时实现上述任一项实施方式所述的方法的步骤。
45、上述说明书实施方式中,首先,获取待检测图像的第一特征点集合与参考图像的第二特征点集合进行特征点匹配,得到第一特征点集合中与第二特征点集合匹配的目标特征点。若目标特征点的数量小于预设数量阈值,确定第二特征点集合中的第二特征点在参考图像上的分布情况。然后,根据分布情况、第一特征点集合以及第二特征点集合进行聚类和特征点配对,得到数量大于预设数量阈值的多个特征点匹配对。最后,基于多个特征点匹配对待检测图像进行校正,得到校正后检测图像。在目标特征点的数量小于预设数量阈值时,表明匹配结果不足以准确地描述待检测图像与参考图像之间的关系。因此,确定第二特征点集合中的第二特征点在参考图像上的分布情况并根据分布情况、第一特征点集合以及第二特征点集合进行聚类和特征点配对,综合考虑特征点的空间分布情况并建立待检测图像与参考图像之间的对应关系,以得到数量大于预设数量阈值的多个特征点匹配。最后,基于数量大于预设数量阈值的多个特征点匹配对待检测图像进行校正,增强校正结果的稳定性和可靠性,提高校正后检测图像的质量,为后续的检测识别任务提供更为精确的图像输入。
1.一种图像校正方法,其特征在于,所述方法包括:
2.根据权利要求1所述的方法,其特征在于,所述根据所述分布情况、所述第一特征点集合以及所述第二特征点集合进行聚类和特征点配对,得到多个特征点匹配对,包括:
3.根据权利要求2所述的方法,其特征在于,所述每个聚类簇在所述第二特征点集合中对应有初始匹配点;所述利用每个聚类簇与所述第二特征点集合中的特征点进行特征点配对,得到所述多个特征点匹配对,包括:
4.根据权利要求3所述的方法,其特征在于,所述第一特征点集合包括对所述待检测图像进行目标检测得到的第一特征点;所述将所述每个聚类簇与所述每个聚类簇对应的初始匹配点进行匹配,得到所述多个特征点匹配对,包括:
5.根据权利要求3所述的方法,其特征在于,所述第一特征点集合包括对所述待检测图像进行目标检测得到的第一特征点;所述将所述每个聚类簇与所述每个聚类簇对应的初始匹配点进行匹配,得到所述多个特征点匹配对,包括以下情况中的至少一个:
6.根据权利要求1至5任一项所述的方法,其特征在于,所述基于所述多个特征点匹配对所述待检测图像进行校正,得到校正后检测图像,包括:
7.根据权利要求1至5任一项所述的方法,其特征在于,通过以下方式获得所述第一特征点集合:
8.根据权利要求1至5任一项所述的方法,其特征在于,通过以下方式确定所述目标特征点:
9.根据权利要求8所述的方法,其特征在于,所述根据所述修正特征点集合与所述第二特征点集合,确定是否从所述修正特征点集合中删除所述待剔除特征点,包括:
10.一种图像校正装置,其特征在于,所述装置包括:
11.一种计算机设备,其特征在于,包括处理器和存储器,所述处理器用于执行所述存储器中的计算机程序,所述计算机程序能够执行权利要求1至9任一所述的方法。
12.一种计算机可读存储介质,其特征在于,用于存储计算机程序,该计算机程序能够执行权利要求1至9任一所述的方法。
13.一种芯片,包括存储单元和处理单元,所述存储单元存储有计算机程序,其特征在于,所述处理单元执行所述计算机程序时实现权利要求1至9中任一项所述的方法的步骤。
