The application relates to a
remote sensing image segmentation method, device and equipment based on spatial affinity learning. The method comprises the following steps: obtaining a multi-level feature map of a
remote sensing training image through a
feature extraction network and a neck network, inputting the multi-level feature map into a head network containing
parallel detection and segmentation branches to obtain hierarchical classification and regression features, performing multi-level aggregation and enhancement on the two types of features through a spatial information enhancement unit, splicing the features into a bounding box feature, generating a
mask feature from the multi-level feature map, guiding the enhancement of the
mask feature through the bounding box feature through a double-flow residual
feature fusion unit to enhance the
spatial perception and semantic coherence of the
mask feature, splicing the mask feature and relative coordinates to input a full
convolution segmentation head to obtain an instance segmentation prediction result, calculating a detection and segmentation
branch loss by taking the prediction result and a horizontal box
label as input, training each module to obtain a segmentation network,
processing a
remote sensing image through the segmentation network to obtain a detection and segmentation result. The method improves the
utilization rate of spatial information of box supervision, optimizes the box supervision remote sensing instance segmentation task, and reduces the gap with the full supervision method.