[发明专利]一种旋转目标检测方法、装置、设备及可读存储介质

权利要求书

1.一种旋转目标检测方法，其特征在于，包括以下步骤：

基于特征提取骨干网络对待定位图像z进行特征提取，得到深度不同的多张第一特征图所述待定位图像z包含多个待定位对象；

基于特征金字塔网络对所述多张第一特征图进行特征聚合，得到感受野不同的多张第二特征图ρ_i(z)；

分别对所述多张第二特征图ρ_i(z)进行上采样，得到具有相同尺寸的多张第三特征图，并将所述多张第三特征图进行相加，得到融合特征图P(z)；

将所述融合特征图P(z)输入至卷积层，得到顶点热力图、日冕中心点热力图、尺寸预测图、偏执预测图和方向预测图；

根据顶点热力图、日冕中心点热力图、尺寸预测图、偏执预测图和方向预测图计算得到所述待定位图像中各个待定位对象的角点坐标。

2.如权利要求1所述的旋转目标检测方法，其特征在于，所述基于特征金字塔网络对所述多张第一特征图进行特征聚合，得到感受野不同的多张第二特征图ρ_i(z)，包括：

基于特征金字塔网络分别对所述多张第一特征图进行卷积操作，得到多张卷积后的特征图

根据以下公式分别对多张卷积后的特征图进行特征相加处理，得到感受野不同的多张第二特征图ρ_i(z)；

3.如权利要求1所述的旋转目标检测方法，其特征在于，所述根据顶点热力图、日冕中心点热力图、尺寸预测图、偏执预测图和方向预测图计算得到所述待定位图像中各个待定位对象的角点坐标，包括：

从顶点热力图中获取多个第一置信度峰值点，并分别将每个第一置信度峰值点作为一个预测顶点，得到多个预测顶点；

从日冕中心点热力图中获取第一待定位对象的第二置信度峰值点，并将所述第二置信度峰值点作为所述第一待定位对象的预测中心点；

从尺寸预测图中获取与预测中心点对应的尺寸预测值，从方向预测图中获取与预测中心点对应的方向预测值，从偏执预测图中分别获取与每个预测顶点对应的第一浮点偏执值、与预测中心点对应的第二浮点偏执值；

根据多个预测顶点、预测中心点、尺寸预测值、方向预测值、多个第一浮点偏执值和第二浮点偏执值计算得到第一待定位对象的角点坐标。

4.如权利要求3所述的旋转目标检测方法，其特征在于，所述根据多个预测顶点、预测中心点、尺寸预测值、方向预测值、多个第一浮点偏执值和第二浮点偏执值计算得到第一待定位对象的角点坐标，包括：

将方向预测值映射为角度预测值；

基于尺寸预测值计算出第一待定位对象在以(0，0)为原点的直角坐标系下的第一角点坐标；

根据预测中心点和第二浮点偏执值计算出第一待定位对象在待定位图像z中的真实中心点；

基于角度预测值、第一角点坐标和真实中心点计算出第一待定位对象在图像坐标系下的第二角点坐标；

根据多个预测顶点、多个第一浮点偏执值、方向预测值和尺寸预测值检测是否存在实际顶点；

若不存在实际顶点，则将所述第二角点坐标作为第一待定位对象的角点坐标。

5.如权利要求4所述的旋转目标检测方法，其特征在于，在所述根据多个预测顶点、多个第一浮点偏执值、方向预测值和尺寸预测值检测是否存在实际顶点的步骤之后，还包括：

若存在实际顶点，则根据实际顶点和真实中心点对所述角度预测值进行更新，得到新的角度预测值；

基于新的角度预测值、第一角点坐标和真实中心点计算出第一待定位对象在图像坐标系下的第三角点坐标；

当检测到所述第三角点坐标与所述第二角点坐标不相等，则将所述第三角点坐标作为第一待定位对象的角点坐标。