[发明专利]目标检测方法、检测模型的训练方法及电子设备

说明书

本发明涉及图像处理技术领域，具体涉及目标检测方法、检测模型的训练方法及电子设备，其中检测方法包括获取待检测图像；将待检测图像输入检测模型中，以得到目标候选区域的位置信息；所述目标候选区域为检测模型输出的所有候选区域的尺寸小于预设值的候选区域；基于目标候选区域的位置信息，在待检测图像中提取预设尺寸的图像，得到待检测子图像；将待检测子图像输入检测模型中，以得到目标候选区域对应的目标的类别。该检测方法仅在待检测图像中提取候选区域的尺寸小于预设尺寸的候选区域进行再次的检测，能够减少数据处理量，提高目标检测的效率。

技术领域

本发明涉及图像处理技术领域，具体涉及目标检测方法、检测模型的训练方法及电子设备。

背景技术

对于图像中目标的检测方法，可以通过人工方式进行检测，但是这种检测方法会受人工主观因素的影响，导致检测的准确性较低。基于此，现有技术中还提出了对图像进行自动检测的方法，以实现对图像中的目标进行自动检测。然而，当待检测图像中存在多尺寸的目标时，要想实现对小尺寸目标进行准确地检测，常采用的方法是将待检测图像分割成多个较小的子图片的方式对小尺寸目标进行检测。

以输电线路检测为例，输电线路巡检是运行与维护输电网的重要工作之一。目前我国输电线路巡视采用直升机巡检、无人机巡检、机器人巡检相结合的方式采集输电线路巡检图片，并使用人工智能方法识别采集的输电线路巡检图像以大大增加查看巡检照片的效率、减轻巡检工人的工作量。然而输电线路巡检图片中存在众多尺寸较小的目标，大部分现有技术直接将原始图片缩放至较大分辨率或将原始图片均匀地分割为多个较小的图片的方式来对尺寸较小的目标进行识别。虽然这些技术能够对小尺寸目标进行较为准确地识别，但是其使用较大分辨率图片或对原始图像直接切割再进行检测的方式需耗费更多的计算资源，导致目标检测的效率较低。

发明内容

有鉴于此，本发明实施例提供了一种目标检测方法、检测模型的训练方法及电子设备，以解决目标检测效率较低的问题。

根据第一方面，本发明实施例提供了一种目标检测方法，包括：

获取待检测图像；

将所述待检测图像输入检测模型中，以得到目标候选区域的位置信息；其中，所述目标候选区域为所述检测模型输出的所有候选区域的尺寸小于预设值的候选区域；

基于所述目标候选区域的位置信息，在所述待检测图像中提取预设尺寸的图像，得到待检测子图像；

将所述待检测子图像输入所述检测模型中，以得到所述目标候选区域对应的目标的类别。

本发明实施例提供的目标检测方法，在利用检测模型得到目标候选区域的位置信息之后，在待检测图像中提取预设尺寸的图像得到待检测子图像，并利用检测模型对待检测子图像进行再次的检测；即，仅在待检测图像中提取候选区域的尺寸小于预设尺寸的候选区域进行再次的检测，能够减少数据处理量，提高目标检测的效率。

结合第一方面，在第一方面第一实施方式中，所述将所述待检测图像输入检测模型中，以得到目标候选区域的位置信息，包括：

利用所述检测模型中的候选区域预测结构预测所述待检测图像中各个目标对应的候选区域；

利用所述检测模型中的区域判断结构判断所述候选区域的尺寸是否小于所述预设值；

当所述候选区域的尺寸小于所述预设值时，利用所述检测模型中的第一区域预测结构预测所述候选区域的第一位置信息，以得到所述目标候选区域的位置信息。

本发明实施例提供的目标检测方法，利用区域判断结构对候选区域的尺寸进行判断，在尺寸小于预设值时仅利用第一区域预设结构预测候选区域的第一位置信息，而不进行类别的预测，在保证检测准确性的前提下，可以提高目标检测的效率。

结合第一方面第一实施方式，在第一方面第二实施方式中，所述第一位置信息为所述目标候选区域的中心位置信息。