[发明专利]ROI提取方法、装置、设备及可读存储介质

说明书

本发明提供一种ROI提取方法、装置、设备及可读存储介质。该方法包括：对原始图像进行感兴趣区域粗定位，确定粗定位边框；在原始图像上截取粗定位边框的每条边对应的子图，得到N张子图，N为粗定位边框包含的边的数量；将每张子图分别输入直线检测模型，得到每张子图对应的直线边缘坐标点集合；根据每张子图对应的直线边缘坐标点集合，进行感兴趣区域精提取。通过本发明，提高了ROI提取精度。

技术领域

本发明涉及自动光学检测技术领域，尤其涉及一种ROI提取方法、装置、设备及可读存储介质。

背景技术

在AOI（Automated Optical Inspection，自动光学检测）领域，由于取像的图像中一般都会包含一定比例的背景信息，而背景区域并不是我们关心的区域，因此第一步都会进行ROI（region of interest，感兴趣区域）的提取，从而减少后续算法的计算量，同时可用有效避免背景区域引入的过检。如何准确提取ROI，是目前亟待解决的技术问题。

发明内容

为解决上述技术问题，本发明提供一种ROI提取方法、装置、设备及可读存储介质。

第一方面，本发明提供一种ROI提取方法，所述ROI提取方法包括：

对原始图像进行感兴趣区域粗定位，确定粗定位边框；

在原始图像上截取粗定位边框的每条边对应的子图，得到N张子图，N为粗定位边框包含的边的数量；

将每张子图分别输入直线检测模型，得到每张子图对应的直线边缘坐标点集合；

根据每张子图对应的直线边缘坐标点集合，进行感兴趣区域精提取。

可选的，所述对原始图像进行感兴趣区域粗定位，确定粗定位边框的步骤包括：

将原始图像缩放到第一预设尺寸，得到第一图像；

将第一图像输入直线检测模型，得到第一图像对应的第一直线边缘坐标点集合；

根据原始图像中与第一直线边缘坐标点集合对应的坐标点集合进行直线拟合；

根据拟合得到的直线，得到粗定位边框。

可选的，所述对原始图像进行感兴趣区域粗定位，确定粗定位边框的步骤包括：

基于预设坐标对原始图像中的目标进行截取，得到目标图像；

将目标图像缩放到第一预设尺寸，得到第二图像；

将第二图像输入直线检测模型，得到第二图像对应的第二直线边缘坐标点集合；

根据原始图像中与第二直线边缘坐标点集合对应的坐标点集合进行直线拟合；

根据拟合得到的直线，得到粗定位边框。

可选的，所述在原始图像上截取粗定位边框的每条边对应的子图的步骤包括：

在原始图像上分别以粗定位边框的每条边上的目标点为中心，截取第二预设尺寸的子图。

可选的，所述粗定位边框的每条边上的目标点为粗定位边框的每条边上的中心点。

可选的，所述将每张子图分别输入直线检测模型，得到每张子图对应的直线边缘坐标点集合的步骤包括：

将每张子图缩放到第一预设尺寸后分别输入直线检测模型，得到每张子图对应的直线边缘坐标点集合。

可选的，所述根据每张子图对应的直线边缘坐标点集合，进行感兴趣区域精提取的步骤包括：

根据每张子图对应的直线边缘坐标点集合确定每张子图对应的直线方程；

根据N张子图对应的N个直线方程，计算得到N个直线交点坐标；

基于所述N个直线交点坐标进行感兴趣区域精提取。

第二方面，本发明还提供一种ROI提取装置，所述ROI提取装置包括：

粗定位模块，用于对原始图像进行感兴趣区域粗定位，确定粗定位边框；