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

权利要求书

1.一种目标检测方法，其特征在于，包括：

获取待检测图像；

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

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

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

其中，所述将所述待检测图像输入检测模型中，以得到目标候选区域的位置信息，包括：

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

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

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

当所述候选区域的尺寸大于或等于所述预设值时，利用所述检测模型中的第二区域预测结构预测所述候选区域对应的目标的类别及第二位置信息。

2.根据权利要求1所述的方法，其特征在于，所述第一位置信息为所述目标候选区域的中心位置信息。

3.根据权利要求1或2所述的方法，其特征在于，所述将所述待检测图像输入检测模型中，以得到目标候选区域的位置信息，还包括：

将所述待检测图像缩放至预设分辨率的待检测图像；

将所述预设分辨率的待检测图像输入所述检测模型中，以得到目标候选区域的位置信息。

4.根据权利要求3所述的方法，其特征在于，所述将所述待检测子图像输入所述检测模型中，以得到所述目标候选区域对应的目标的类别，包括：

将所述待检测子图像缩放至所述预设分辨率的待检测子图像；

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

5.根据权利要求1所述的方法，其特征在于，所述待检测图像为输电线路图像。

6.一种检测模型的训练方法，其特征在于，包括：

获取带有标注信息的样本图像；其中，所述标注信息为所述样本图像中各个目标对应的类别及目标区域的位置信息，所述目标区域的尺寸小于预设值；

将所述样本图像输入初始检测模型中，以得到预测区域的位置信息；

基于所述样本图像中标注出的目标区域的位置信息与所述预测区域的位置信息，对所述初始检测模型中的参数进行更新，以得到所述检测模型；

其中，所述检测模型包括候选区域预测结构、区域判断结构以及第一区域预测结构，所述候选区域预测结构用于预测待检测图像中各个目标的候选区域，所述区域判断结构用于判断所述各个目标的候选区域的尺寸是否小于预设值以确定目标候选区域，所述第一区域预测结构用于确定所述目标候选区域的位置信息；当所述候选区域的尺寸小于所述预设值时，利用所述检测模型中的第一区域预测结构预测所述候选区域的第一位置信息，以得到所述目标候选区域的位置信息；当所述候选区域的尺寸大于或等于所述预设值时，利用所述检测模型中的第二区域预测结构预测所述候选区域对应的目标的类别及第二位置信息。

7.一种电子设备，其特征在于，包括：

存储器和处理器，所述存储器和所述处理器之间互相通信连接，所述存储器中存储有计算机指令，所述处理器通过执行所述计算机指令，从而执行权利要求1-5中任一项所述的目标检测方法，或权利要求6所述的检测模型的训练方法。

8.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有计算机指令，所述计算机指令用于使所述计算机执行权利要求1-5中任一项所述的目标检测方法，或权利要求6所述的检测模型的训练方法。