The invention discloses a
remote sensing image omni-directional target detection
algorithm based on a multilayer feature interaction
pyramid and a lightweight enhanced detection head, and belongs to the field of
computer vision. The method comprises the following steps: 1, preprocessing a
remote sensing image
data set; 2, building a
remote sensing image omnibearing target detection model: designing a multi-layer feature interaction
pyramid, obtaining an intermediate feature map and a fusion feature map by aggregating multi-layer feature maps, realizing cross-layer
feature fusion, preventing information interaction from being limited between adjacent
layers, generating rotation-
invariant feature representation, and enhancing feature information of a rotating target; a lightweight enhanced detection head is constructed, the parameter quantity of the model is reduced by adopting a shared enhanced
convolution strategy, and feature information is extracted through central difference
convolution, so that the model can capture detail features of a rotating target; and the complexity of the model is reduced by adopting a Lamp
pruning method on the basis that the precision is not lost. And 3, constructing a
loss function of the model, and introducing a KLD
divergence loss function to solve the problem of periodic angle change in rotating target detection. And 4, iteratively training the model until the model reaches convergence, and obtaining the
optimal weight. And 5, testing the
test set according to the obtained
optimal weight to obtain an
evaluation result. According to the method, the parameter quantity and the calculation complexity of the model are reduced while the detection precision of the rotating target is ensured.