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

权利要求书

1.一种图像检测方法，其特征在于，所述方法包括：

获取待识别图像；

利用yolov5网络获取所述待识别图像对应的第一检测框；

判断所述第一检测框是否满足预设条件；

在所述第一检测框不满足预设条件的情况下，利用Dbnet网络获取所述待识别图像对应的第二检测框；

基于所述第一检测框和第二检测框确定所述待识别图像对应的第三检测框。

2.根据权利要求1所述的图像检测方法，其特征在于，所述第一检测框以第一检测框对角线上的第一坐标和第二坐标表示；

判断所述第一检测框是否满足预设条件，包括：

基于所述第一坐标和第二坐标确定所述第一检测框的长宽比；

判断所述长宽比是否大于第一阈值；

若是，则确定所述第一检测框满足预设条件；若否，则确定所述第一检测框不满足第一预设条件。

3.根据权利要求1所述的图像检测方法，其特征在于，基于所述第一检测框和第二检测框确定所述待识别图像对应的第三检测框，包括：

判断所述第一检测框和第二检测框是否有交集；

在所述第一检测框和第二检测框有交集的情况下，基于有交集的第一检测框和第二检测框确定所述待识别图像对应的第三检测框。

4.根据权利要求3所述的图像检测方法，其特征在于，在所述第一检测框和第二检测框有交集的情况下，基于有交集的第一检测框和第二检测框确定所述待识别图像对应的第三检测框，包括：在与所述第一检测框有交集的第二检测框有多个时，基于置信度最高的第二检测框和所述第一检测框确定所述待识别图像对应的第三检测框。

5.根据权利要求3或4所述的图像检测方法，其特征在于，基于有交集的第一检测框和第二检测框确定所述待识别图像对应的第三检测框，包括：

确定是否修正所述第二检测框；

若是，则基于所述第一检测框修正所述第二检测框，得到所述第三检测框；若否，将所述第二检测框作为第三检测框。

6.根据权利要求5所述的图像检测方法，其特征在于，所述第二检测框以四个顶点坐标表示，分别为第三坐标、第四坐标、第五坐标和第六坐标；确定是否修正所述第二检测框，包括：

确定所述第三坐标的横坐标与第一坐标的横坐标之间的第一差值，确定所述第六坐标的横坐标与第一坐标的横坐标的第二差值，确定所述第四坐标的横坐标与第二坐标的横坐标之间的第三差值，确定所述第五坐标的横坐标与第二坐标的横坐标之间的第四差值；

判断所述第一差值与第二差值是否均大于第二阈值，和/或所述第三差值与第四差值是否均大于第二阈值；

若是，则确定修正所述第二检测框；若否，则确定不修正所述第二检测框。

7.根据权利要求6所述的图像检测方法，其特征在于，基于所述第一检测框修正所述第二检测框，包括：

在所述第一差值与第二差值均大于第二阈值的情况下，将所述第三坐标的横坐标和所述第六坐标的横坐标替换为所述第一坐标的横坐标；和/或

在所述第三差值与第四差值均大于第二阈值的情况下，将所述第四坐标的横坐标和所述第五坐标的横坐标替换为所述第二坐标的横坐标。

8.一种图像检测装置，其特征在于，所述装置包括：

第一获取模块，被配置为获取待识别图像；

第二获取模块，被配置为利用yolov5网络获取所述待识别图像对应的第一检测框；

判断模块，被配置为判断所述第一检测框是否满足预设条件；

第三获取模块，被配置为在所述第一检测框不满足预设条件的情况下，利用Dbnet网络获取所述待识别图像对应的第二检测框；

确定模块，被配置为基于所述第一检测框和第二检测框确定所述待识别图像对应的第三检测框。

9.一种电子设备，包括处理器和存储装置，所述存储装置适于存储多条程序代码，其特征在于，所述程序代码适于由所述处理器加载并运行以执行权利要求1至7中任一项所述的图像检测方法。

10.一种计算机可读存储介质，其中存储有多条程序代码，其特征在于，所述程序代码适于由处理器加载并运行以执行权利要求1至7中任一项所述的图像检测方法。