[发明专利]图像检测方法、装置、电子设备及存储介质

说明书

本发明涉及图像检测技术领域，具体提供一种图像检测方法、装置、电子设备及存储介质，旨在解决现有的图像检测方法对应的图像检测精度较低的技术问题。为此目的，本发明的图像检测方法包括：获取待识别图像；利用yolov5网络获取待识别图像对应的第一检测框；判断第一检测框是否满足预设条件；在第一检测框不满足预设条件的情况下，利用Dbnet网络获取待识别图像对应的第二检测框；基于第一检测框和第二检测框确定待识别图像对应的第三检测框。如此，提高了检测框的检测精度。

技术领域

本发明涉及图像检测技术领域，具体提供一种图像检测方法、装置、电子设备及存储介质。

背景技术

目前，银行卡的检测技术比较成熟，现有技术中针对一般场景下的银行卡检测存在较多方案，但是在例如光线暗、拍摄图像时角度倾斜严重、拍摄的时候卡片在图像中占比较小以及卡片的磨损情况比较严重等情况下，图像检测精度较低。

相应地，本领域需要一种新的图像检测方案来解决上述问题。

发明内容

为了克服上述缺陷，提出了本发明，以提供解决或至少部分地解决上述技术问题。本发明提供了一种图像检测方法、装置、电子设备及存储介质。

在第一方面，本发明提供一种图像检测方法，所述方法包括：获取待识别图像；利用yolov5网络获取所述待识别图像对应的第一检测框；判断所述第一检测框是否满足预设条件；在所述第一检测框不满足预设条件的情况下，利用Dbnet网络获取所述待识别图像对应的第二检测框；基于所述第一检测框和第二检测框确定所述待识别图像对应的第三检测框。

在一个实施方式中，所述第一检测框以第一检测框对角线上的第一坐标和第二坐标表示；判断所述第一检测框是否满足预设条件，包括：基于所述第一坐标和第二坐标确定所述第一检测框的长宽比；判断所述长宽比是否大于第一阈值；若是，则确定所述第一检测框满足预设条件；若否，则确定所述第一检测框不满足第一预设条件。

在一个实施方式中，基于所述第一检测框和第二检测框确定所述待识别图像对应的第三检测框，包括：判断所述第一检测框和第二检测框是否有交集；在所述第一检测框和第二检测框有交集的情况下，基于有交集的第一检测框和第二检测框确定所述待识别图像对应的第三检测框。

在一个实施方式中，在所述第一检测框和第二检测框有交集的情况下，基于有交集的第一检测框和第二检测框确定所述待识别图像对应的第三检测框，包括：在与所述第一检测框有交集的第二检测框有多个时，基于置信度最高的第二检测框和所述第一检测框确定所述待识别图像对应的第三检测框。

在一个实施方式中，基于有交集的第一检测框和第二检测框确定所述待识别图像对应的第三检测框，包括：确定是否修正所述第二检测框；若是，则基于所述第一检测框修正所述第二检测框，得到所述第三检测框；若否，将所述第二检测框作为第三检测框。

在一个实施方式中，所述第二检测框以四个顶点坐标表示，分别为第三坐标、第四坐标、第五坐标和第六坐标；确定是否修正所述第二检测框，包括：确定所述第三坐标的横坐标与第一坐标的横坐标之间的第一差值，确定所述第六坐标的横坐标与第一坐标的横坐标的第二差值，确定所述第四坐标的横坐标与第二坐标的横坐标之间的第三差值，确定所述第五坐标的横坐标与第二坐标的横坐标之间的第四差值；判断所述第一差值与第二差值是否均大于第二阈值，和/或所述第三差值与第四差值是否均大于第二阈值；若是，则确定修正所述第二检测框；若否，则确定不修正所述第二检测框。

在一个实施方式中，基于所述第一检测框修正所述第二检测框，包括：在所述第一差值与第二差值均大于第二阈值的情况下，将所述第三坐标的横坐标和所述第六坐标的横坐标替换为所述第一坐标的横坐标；和/或在所述第三差值与第四差值均大于第二阈值的情况下，将所述第四坐标的横坐标和所述第五坐标的横坐标替换为所述第二坐标的横坐标。

在第二方面，本发明提供一种图像检测装置，所述装置包括：