The invention discloses a building extraction method based on a high-resolution network, and belongs to the field of
image processing and
machine vision detection. In order to solve the problems of void phenomenon, poor boundary precision, missing extraction and mistaken extraction of small-sized buildings and the like existing in high-resolution
remote sensing image building extraction of an existing
deep learning method, the method adopts a multi-stage network optimization strategy: firstly, obtaining a high-resolution building
data set; and expanding the
training set by adopting a data enhancement technology of sample
random rotation transformation,
Gaussian filtering boundary enhancement, HSV-based
color space transformation and standard deviation-added random
Gaussian noise. Sequentially integrating object context representation (OCR) modules on the HRNet
backbone network to reduce internal holes; introducing a boundary refinement post-
processing Segfix module to carry out direction-guided pixel-level correction on an output boundary so as to optimize the building boundary; a rotary variable-size window attention (RVSA) mechanism is embedded in a down-sampling layer, and the capacity of capturing
small target features is enhanced through a dynamically changing window. And finally, end-to-end training is carried out on the constructed model by using an enhanced
data set, accurate and complete extraction of the building is realized, and an efficient and reliable solution is provided for scene applications such as
urban planning and
disaster assessment.