[发明专利]一种基于边缘提取的图像识别方法及系统

说明书

本发明涉及一种基于边缘提取的图像识别方法，属于图像识别技术领域，该方法包括：对待识别图像进行灰度化处理，获得第一灰度图；对第一灰度图进行降噪和平滑处理，获得第二灰度图；对第二灰度图进行边缘检测，获得边缘信息图；对边缘信息图进行二值化处理，获得二值化图像；通过遍历二值化图像确定目标区域的边界框，获得带有边界框的二值化图像；根据带有边界框的二值化图像中边界框对待识别图像进行裁剪，获得裁剪后的图像；利用裁剪后的图像进行图像识别。本发明根据边界框对待识别图像进行裁剪，从而实现低成本、高效率地对感兴趣区域的提取，解决了背景信息过多对图像识别算法性能的不良影响，有助于提高图像识别的精度。

技术领域

本发明涉及图像识别技术领域，特别是涉及一种基于边缘提取的图像识别方法及系统。

背景技术

目前现有的图像识别方法，往往对待识别图像直接进行缩放，再将缩放的图像输入到图像识别模型中进行识别。当待识别图像中目标较小，背景信息过多，图像识别模型通常无法正确识别图像。虽然结合现有的感兴趣区域搜索方法进行图像识别或使用目标检测进行图像识别可以解决这一问题，但此类方法具有较高的计算量与算法复杂度，从而带来了很大的时间消耗。

发明内容

本发明的目的是提供一种基于边缘提取的图像识别方法，降低图像的识别成本，提高了图像识别精度。

为实现上述目的，本发明提供了如下方案：

一种基于边缘提取的图像识别方法，包括：

对待识别图像进行灰度化处理，获得第一灰度图；

对所述第一灰度图进行降噪和平滑处理，获得第二灰度图；

对所述第二灰度图进行边缘检测，获得边缘信息图；

对所述边缘信息图进行二值化处理，获得二值化图像；

通过遍历所述二值化图像确定目标区域的边界框，获得带有边界框的二值化图像；

根据所述带有边界框的二值化图像中边界框对所述待识别图像进行裁剪，获得裁剪后的图像；

利用所述裁剪后的图像进行图像识别。

可选地，所述通过遍历所述二值化图像确定目标区域的边界框，获得带有边界框的二值化图像，具体包括：

从第一行开始，按照从左到右，从上到下的顺序，逐行遍历所述二值化图像的像素点，当查找到非零像素点时，结束当前遍历，将当前非零像素点作为所述边界框左上角的横坐标；

从第一列开始，按照从上到下，从左到右的顺序，逐列遍历所述二值化图像的像素点，当查找到非零像素点时，结束当前遍历，将当前非零像素点作为所述边界框左上角的纵坐标；

从最后一行开始，按照从右到左，从下到上的顺序，逐行遍历所述二值化图像的像素点，当查找到非零像素点时，结束当前遍历，将当前非零像素点作为所述边界框右下角的横坐标；

从最后一列开始，按照从下到上，从右到左的顺序，逐列遍历所述二值化图像的像素点，当查找到非零像素点时，结束当前遍历，将当前非零像素点作为所述边界框右下角的纵坐标。

可选地，所述对所述第二灰度图进行边缘检测，获得边缘信息图，具体包括：

采用prewitt算子对所述第二灰度图进行边缘检测，获得边缘信息图。

可选地，所述利用所述裁剪后的图像进行图像识别，具体包括：

将所述裁剪后的图像缩放至设定尺寸，获得设定尺寸的图像；

将所述设定尺寸的图像输入图像识别模型进行目标识别。

可选地，所述将所述裁剪后的图像缩放至设定尺寸，获得设定尺寸的图像，具体包括：