[发明专利]目标检测及目标检测网络的训练方法、装置及设备

说明书

公开了一种目标检测及目标检测网络的训练方法、装置及设备。该目标检测方法包括：获得输入图像的特征数据；根据所述特征数据，确定所述输入图像的多个候选边界框；根据所述特征数据，获得所述输入图像的前景分割结果，其中，前景分割结果包含指示所述输入图像的多个像素中每个像素是否属于前景的指示信息；根据所述多个候选边界框与所述前景分割结果，得到所述输入图像的目标检测结果。

技术领域

本公开涉及图像处理技术领域，尤其涉及一种目标检测及目标检测网络的训练方法、装置及设备。

背景技术

目标检测是计算机视觉领域的重要问题，尤其对于飞机、舰船等军事目标的检测，由于其具有影像尺寸大、目标尺寸小的特点，检测难度大。而且，对于具有密集排列状态的舰船等目标，目前的目标检测方法的检测精度还有待进一步提升。

发明内容

本公开实施例提供了一种目标检测及目标检测网络的训练方法、装置及设备。

第一方面，提供一种目标检测方法，包括：

获得输入图像的特征数据；

根据所述特征数据，确定所述输入图像的多个候选边界框；

根据所述特征数据，获得所述输入图像的前景分割结果，其中，前景分割结果包含指示所述输入图像的多个像素中每个像素是否属于前景的指示信息；

根据所述多个候选边界框与所述前景分割结果，得到所述输入图像的目标检测结果。

结合本公开提供的任一实施方式，所述根据所述多个候选边界框与所述前景分割结果，得到所述输入图像的目标检测结果，包括：

根据所述多个候选边界框中每个候选边界框与所述前景分割结果对应的前景图像区域之间的重叠区域，从多个候选边界框中选取至少一个目标边界框；

基于所述至少一个目标边界框，得到所述输入图像的目标检测结果。

结合本公开提供的任一实施方式，所述根据所述多个候选边界框中每个候选边界框与所述前景分割结果对应的前景图像区域之间的重叠区域，从多个候选边界框中选取至少一个目标边界框，包括：

将所述多个候选边界框中与所述前景图像区域之间的重叠区域在整个候选边界框中所占的比例大于第一阈值的候选边界框作为所述目标边界框。

结合本公开提供的任一实施方式，所述至少一个目标边界框包括第一边界框和第二边界框，所述基于所述至少一个目标边界框，得到所述输入图像的目标检测结果，包括：

基于所述第一边界框和所述第二边界框之间的夹角，确定所述第一边界框和所述第二边界框的重叠参数；

基于所述第一边界框和所述第二边界框的重叠参数，确定所述第一边界框和所述第二边界框所对应的目标对象位置。

结合本公开提供的任一实施方式，所述基于所述第一边界框和所述第二边界框之间的夹角，确定所述第一边界框和所述第二边界框的重叠参数，包括：

根据所述第一边界框和所述第二边界框之间的夹角，获得角度因子；

根据所述第一边界框和所述第二边界框之间的交并比和所述角度因子，获得所述重叠参数。

结合本公开提供的任一实施方式，所述重叠参数为所述交并比与所述角度因子的乘积，其中，所述角度因子随着所述第一边界框和所述第二边界框之间的角度的增大而增大。

结合本公开提供的任一实施方式，在所述交并比保持一定的条件下，所述重叠参数随着所述第一边界框和所述第二边界框之间的角度的增大而增大。

结合本公开提供的任一实施方式，在所述重叠参数大于第二阈值的情况下，将所述第一边界框和所述边界框中的其中一个边界框作为目标对象位置。