[发明专利]图像检测模型的训练方法、图像检测方法、装置及设备

说明书

本申请公开了一种图像检测模型的训练方法、图像检测方法、装置及设备，属于人工智能技术领域。方法包括：获取样本图像以及样本图像的标注几何图形参数，样本图像的标注几何图形参数是通过标注得到的样本图像中的几何图形的参数信息；根据第一网络模型获取样本图像的预测几何图形参数，样本图像的预测几何图形参数是通过预测得到的样本图像中的几何图形的参数信息；基于样本图像的标注几何图形参数和预测几何图形参数，对第一网络模型进行调整，得到图像检测模型。本申请实施例通过几何图形的参数信息表征几何图形，实现快速地确定出图像中的几何图形，简化了图像检测步骤，提高了图像检测效率。

技术领域

本申请实施例涉及人工智能技术领域，特别涉及一种图像检测模型的训练方法、图像检测方法、装置及设备。

背景技术

随着计算机的发展，越来越多的数据以图像的形式存在于计算机中，并通过计算机实现对图像的处理。在图像处理技术中，图像检测技术是一项重要的技术，图像检测技术能够检测出图像中的几何图形，该几何图形包括直线、椭圆、平面等。

以检测图像中的目标平面为例，相关技术中，获取样本图像的平面以及该平面是否属于目标平面的标注信息，利用网络模型获取平面属于目标平面的概率信息，基于标注信息和概率信息获取到图像检测模型。通过这种方式获取到的图像检测模型在检测目标图像的目标平面时，需要先获取目标图像的某一个平面，将该平面输入图像检测模型，由图像检测模型确定该平面是否为目标平面，若该平面不为目标平面，则获取目标图像的另一个平面，重新由图像检测模型确定该平面是否为目标平面。

上述方式获取到的图像检测模型在检测图像的目标平面时，需要不断地获取图像的不同平面，并将平面输入图像检测模型，直至图像检测模型确定出目标平面为止。这种方式较为繁琐，导致图像检测的效率低下。

发明内容

本申请实施例提供了一种图像检测模型的训练方法、图像检测方法、装置及设备，可用于解决相关技术训练出的图像检测模型在确定平面时方式繁琐、图像检测效率低下的问题，所述技术方案包括如下内容。

一方面，本申请实施例提供了一种图像检测模型的训练方法，所述方法包括：

获取样本图像以及所述样本图像的标注几何图形参数，所述样本图像的标注几何图形参数是通过标注得到的所述样本图像中的几何图形的参数信息；

根据第一网络模型获取所述样本图像的预测几何图形参数，所述样本图像的预测几何图形参数是通过预测得到的所述样本图像中的几何图形的参数信息；

基于所述样本图像的标注几何图形参数和所述预测几何图形参数，对所述第一网络模型进行调整，得到图像检测模型。

另一方面，本申请实施例提供了一种图像检测方法，所述方法包括：

获取目标图像；

根据图像检测模型获取所述目标图像的预测几何图形参数，所述图像检测模型是按照上述任一所述的图像检测模型的训练方法得到的，所述目标图像的预测几何图形参数是通过预测得到的所述目标图像中的几何图形的参数信息；

基于所述目标图像的预测几何图形参数，确定所述目标图像中的目标几何图形。

另一方面，本申请实施例提供了一种图像检测模型的训练装置，所述装置包括：

第一获取模块，用于获取样本图像以及所述样本图像的标注几何图形参数，所述样本图像的标注几何图形参数是通过标注得到的所述样本图像中的几何图形的参数信息；

第二获取模块，用于根据第一网络模型获取所述样本图像的预测几何图形参数，所述样本图像的预测几何图形参数是通过预测得到的所述样本图像中的几何图形的参数信息；

调整模块，用于基于所述样本图像的标注几何图形参数和所述预测几何图形参数，对所述第一网络模型进行调整，得到图像检测模型。