CN112070037A - Road extraction method, device, medium and equipment based on remote sensing image - Google Patents

Road extraction method, device, medium and equipment based on remote sensing image Download PDF

Info

Publication number
CN112070037A
CN112070037A CN202010953874.4A CN202010953874A CN112070037A CN 112070037 A CN112070037 A CN 112070037A CN 202010953874 A CN202010953874 A CN 202010953874A CN 112070037 A CN112070037 A CN 112070037A
Authority
CN
China
Prior art keywords
remote sensing
road
image
sensing image
distance
Prior art date
Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
Granted
Application number
CN202010953874.4A
Other languages
Chinese (zh)
Other versions
CN112070037B (en
Inventor
陈若男
彭玲
刘玉菲
吕蓓茹
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Aerospace Information Research Institute of CAS
Original Assignee
Aerospace Information Research Institute of CAS
Priority date (The priority date is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the date listed.)
Filing date
Publication date
Application filed by Aerospace Information Research Institute of CAS filed Critical Aerospace Information Research Institute of CAS
Priority to CN202010953874.4A priority Critical patent/CN112070037B/en
Publication of CN112070037A publication Critical patent/CN112070037A/en
Application granted granted Critical
Publication of CN112070037B publication Critical patent/CN112070037B/en
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Images

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V20/00Scenes; Scene-specific elements
    • G06V20/10Terrestrial scenes
    • G06V20/182Network patterns, e.g. roads or rivers
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/21Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
    • G06F18/214Generating training patterns; Bootstrap methods, e.g. bagging or boosting
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/24Classification techniques
    • G06F18/241Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches
    • G06F18/2415Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches based on parametric or probabilistic models, e.g. based on likelihood ratio or false acceptance rate versus a false rejection rate
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06VIMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
    • G06V10/00Arrangements for image or video recognition or understanding
    • G06V10/40Extraction of image or video features
    • G06V10/44Local feature extraction by analysis of parts of the pattern, e.g. by detecting edges, contours, loops, corners, strokes or intersections; Connectivity analysis, e.g. of connected components

Abstract

Provided are a method, device, medium and equipment for extracting a road based on a remote sensing image. The method comprises the following steps: drawing a road in the remote sensing image, and converting the road into a two-classification image; dividing the remote sensing image and the binary image to manufacture a remote sensing image sample and a binary image sample; carrying out edge symbol distance conversion on the binary image samples to obtain edge symbol distance image samples corresponding to the binary image samples one to one; training a road extraction model by using the corresponding remote sensing image sample, the binary image sample and the edge symbol distance image sample; and carrying out road extraction on the remote sensing image by using the trained road extraction model. When the road is extracted, the road continuity performance is obviously improved, and most of the shielding problems can be overcome.

Description

Road extraction method, device, medium and equipment based on remote sensing image
Technical Field
The invention belongs to the technical field of information, and particularly relates to a road extraction method, a road extraction device, a road extraction medium and road extraction equipment based on remote sensing images.
Background
Road information is one of the most important geographic information elements, and plays an important role in applications such as automatic driving and disaster emergency response. With the development of remote sensing technology and the maturity of deep learning technology, remote sensing image road extraction algorithms with high spatial and temporal resolution based on cnn are endless, and allow large-scale monitoring of roads. Therefore, the remote sensing image data quickly becomes an important data source for automatically extracting the road network, and the research on the algorithm for automatically extracting the road from the remote sensing image becomes a focus.
However, the current research mainly focuses on road extraction in urban areas, and since roads in remote areas are often narrow and variable in width, and have problems of serious tree crowns, shadow occlusion and the like, the road extraction algorithms for urban areas are often poor in effect in remote areas, and particularly the problem of road discontinuity and breakage is serious.
Disclosure of Invention
The present invention is directed to solving the problems described above. Specifically, the invention provides a road extraction method, a road extraction device, a road extraction medium and road extraction equipment based on remote sensing images.
According to a first aspect of the present disclosure, there is provided a method for extracting a road based on a remote sensing image, including:
drawing a road in the remote sensing image, and converting the road into a two-classification image;
dividing the remote sensing image and the binary image to manufacture a remote sensing image sample and a binary image sample, wherein the remote sensing image sample and the binary image sample correspond to each other one by one;
carrying out edge symbol distance transformation on the binary image samples to obtain edge symbol distance image samples corresponding to the binary image samples one to one;
training a road extraction model by using the corresponding remote sensing image sample, the binary image sample and the edge symbol distance image sample;
and carrying out road extraction on the remote sensing image by using the trained road extraction model.
Performing edge-symbol distance transformation on the binary image samples comprises:
determining in the two-class image sampleDistance D from each pixel point to the closest point on the road edgeiBased on DiAnd determining the edge symbol distance of all the pixel points.
Figure BDA0002677947010000021
Wherein x isiIs the pixel location, xjOn the road edge and xiThe nearest pixel position, ED, is the Euclidean distance.
Based on DiDetermining the edge symbol distances of all the pixel points comprises:
edge symbol distance BSDiThen, then
Figure BDA0002677947010000022
Wherein Htanh is a HardTanh function, alpha is a proportionality coefficient, F is a foreground area, and B is a background area.
The road extraction model comprises a ResNet frame, a distance regression task branch and a classification task branch.
The method for training the road extraction model by using the corresponding remote sensing image sample, the binary image sample and the edge symbol distance image sample comprises the following steps:
inputting a remote sensing image sample into a ResNet frame to obtain a remote sensing image characteristic diagram;
inputting the remote sensing image feature map into a distance regression task branch, and calculating a first Loss through a first Loss function between the output result of the distance regression task branch and the edge symbol distance image sample;
and inputting the shallow features and the deep features obtained from the distance regression task branch into a classification task branch, and calculating a second Loss through a second Loss function by using the output result of the classification task branch and the binary image sample.
The road extraction method based on the remote sensing image further comprises the following steps:
and determining the final Loss based on the first Loss and the second Loss, and training the road extraction model by using the final Loss.
According to another aspect of the present disclosure, there is provided a remote sensing image-based road extraction device, including:
the second classification image making module is used for drawing the road in the remote sensing image and converting the road into a second classification image;
the sample manufacturing module is used for segmenting the remote sensing image and the binary image to manufacture a remote sensing image sample and a binary image sample, and the remote sensing image sample corresponds to the binary image sample one by one;
the edge symbol distance transformation module is used for carrying out edge symbol distance transformation on the binary image samples to obtain edge symbol distance image samples which correspond to the binary image samples one by one;
the model training module is used for training a road extraction model by using the corresponding remote sensing image sample, the binary image sample and the edge symbol distance image sample;
and the road extraction module is used for extracting the road of the remote sensing image by using the trained road extraction model.
According to another aspect herein, there is provided a computer readable storage medium having stored thereon a computer program which, when executed, performs the steps of a method for remote sensing imagery-based road extraction.
According to another aspect of the present document, there is provided a computer device comprising a processor, a memory and a computer program stored on said memory, the steps of the method for remote sensing image based road extraction being implemented when the computer program is executed by the processor.
The method comprises the steps of carrying out edge symbol distance transformation on a binary image of a remote sensing image to obtain an edge symbol distance image sample, and training a multi-branch road extraction model by using the remote sensing image sample, the binary image sample and the edge symbol distance image sample. The real-value distance information in the edge symbol distance image can promote the model to better identify the road boundary, and better classify road pixels by learning the distance characteristics to the boundary, rather than learning only spectral characteristics, so as to achieve the purpose of correcting shape deformity or breaking the road.
Other characteristic features and advantages of the invention will become apparent from the following description of exemplary embodiments, which is to be read in connection with the accompanying drawings.
Drawings
The accompanying drawings, which are incorporated in and constitute a part of the specification, illustrate embodiments of the invention and together with the description, serve to explain the principles of the invention. In the drawings, like reference numerals are used to indicate like elements. The drawings in the following description are directed to some, but not all embodiments of the invention. For a person skilled in the art, other figures can be derived from these figures without inventive effort.
Fig. 1 is a flowchart illustrating a method for extracting a road based on a remote sensing image according to an exemplary embodiment.
Fig. 2 is a schematic diagram of a remote sensing image according to an exemplary embodiment.
FIG. 3 is a diagram illustrating a binary image according to an exemplary embodiment.
FIG. 4 is a schematic diagram illustrating an edge symbol distance image in accordance with an exemplary embodiment.
FIG. 5 is a diagram illustrating a road extraction model according to an exemplary embodiment.
FIG. 6 is a diagram illustrating predicted results of a road extraction model according to an exemplary embodiment.
Fig. 7 is a block diagram illustrating a road extraction device based on a remote sensing image according to an exemplary embodiment.
Fig. 8 is a block diagram illustrating a computer device for remote sensing image-based road extraction according to an exemplary embodiment.
Detailed Description
In order to make the objects, technical solutions and advantages of the embodiments of the present invention clearer, the technical solutions in the embodiments of the present invention will be clearly and completely described below with reference to the drawings in the embodiments of the present invention, and it is obvious that the described embodiments are some, but not all, embodiments of the present invention. All other embodiments, which can be derived by a person skilled in the art from the embodiments given herein without making any creative effort, shall fall within the protection scope of the present invention. It should be noted that the embodiments and features of the embodiments in the present application may be arbitrarily combined with each other without conflict.
Fig. 1 is a flowchart illustrating a method for extracting a road based on a remote sensing image according to an exemplary embodiment. Referring to fig. 1, the method for extracting a road based on a remote sensing image includes:
step S11: drawing a road in the remote sensing image, and converting the road into a two-classification image;
step S12: dividing the remote sensing image and the binary image to manufacture a remote sensing image sample and a binary image sample, wherein the remote sensing image sample and the binary image sample correspond to each other one by one;
step S13: carrying out edge symbol distance transformation on the binary image samples to obtain edge symbol distance image samples corresponding to the binary image samples one to one;
step S14: training a road extraction model by using the corresponding remote sensing image sample, the binary image sample and the edge symbol distance image sample;
step S15: and carrying out road extraction on the remote sensing image by using the trained road extraction model.
In step S11, the road object in the remote sensing image is rendered, and a rendering tool such as Arcgis, labelme, etc. may be selected. Taking Arcgis as an example, a surface layer element is newly built in a remote sensing image, a geographical coordinate system which is the same as that of the original remote sensing image is given, then, a road in the remote sensing image is sketched through manual visual interpretation by an editing tool, a proper scaling is selected, drawing is carried out along edge pixels, and finally a vector file in a shape file format is generated. And then, converting the vector diagram into a grid diagram to obtain a binary grid image, namely a two-class image, wherein 0 in the two-class image is a background and 1 is a road.
In step S12, a sample segmentation rule is set, including both the size of a single sample and the overlap ratio between adjacent samples. According to the set sample segmentation rule, the remote sensing image and the two-classification image are segmented, and a plurality of segmented remote sensing image samples and a plurality of two-classification image samples can be obtained and used as training samples of a subsequent road extraction model.
In step S13, edge-sign distance conversion is performed on the binary image samples to obtain edge-sign distance image samples corresponding to the binary image samples one to one. When only two classification images are used for training the model, the depth classification model determines whether one pixel belongs to a road class or a non-road class, and the high spatial distribution characteristic correlation among road pixels can be ignored, so that the road extraction result often enables the blocked part of pixels to be classified as non-roads, and the problem of road breakage discontinuity is caused.
Thus introducing edge-symbol distance herein, in one embodiment, edge-symbol distance transforming the binary image samples comprises:
determining a distance D from each pixel point in the two classified image samples to a closest point located on a road edgeiBased on DiAnd determining the edge symbol distance of all the pixel points.
Given a pixel position x, inspired by the symbol distance transform SDTiIts signed distance value SDTiComprises the following steps:
Figure BDA0002677947010000051
where ED is the euclidean distance and F and B are the foreground and background regions, respectively. Symbolic distance value SDTiRepresenting the distance of a pixel to its closest different class of pixels.
Thus, the distance from a given pixel to the nearest pixel located on the edge of the road can be defined as Di
Figure BDA0002677947010000061
Wherein x isiIs the pixel location, xjOn the road edge and xiThe nearest pixel position, ED, is the Euclidean distance.
Edge symbol distance BSDi
Figure BDA0002677947010000062
Wherein Htanh is a HardTanh function, alpha is a proportionality coefficient, F is a foreground area, and B is a background area. Because the calculation is carried out on the binary image, the foreground area is a road, and the background area is a non-road.
In the above formula, the distance DiThe scaling is realized through logarithmic mapping, so that the sensitivity of the edge symbol distance to the distance change is reduced along with the increase of the distance between the pixel point and the road edge, the attention is concentrated on the road edge and the adjacent area, and the attention to the background area far away from the road edge is weakened. The distance values are then normalized to [ -1,1] using the HardTanh function and the scaling factor α]The training speed of the model is increased, and meanwhile, the attention to background pixels far away from the road is further weakened, so that some noises are filtered. Giving the road pixels an edge-symbol distance of positive, the non-road pixels an edge-symbol distance of negative, and the road boundary pixels an edge-symbol distance value of zero will effectively help the model to better distinguish between roads and non-roads. Fig. 2 is a schematic diagram of a remote sensing image according to an exemplary embodiment. FIG. 3 is a diagram illustrating a binary image according to an exemplary embodiment. FIG. 4 is a schematic diagram illustrating an edge symbol distance image in accordance with an exemplary embodiment. It can be seen that the remote sensing image schematic diagram, the binary image and the edge symbol distance image have a one-to-one correspondence relationship.
In step S14, the road extraction model is trained using the corresponding remote sensing image sample, binary image sample, and edge sign distance image sample. And (4) taking the remote sensing image, the two-class image and the edge symbol distance image which are manufactured in the steps S11 to S13 as training samples to train the road extraction model.
In one embodiment, the road extraction model includes a ResNet framework, distance regression task branches, and classification task branches.
FIG. 5 is a diagram illustrating a road extraction model according to an exemplary embodiment. Referring to fig. 5, there are shown a ResNet frame 51, a distance regression task branch 52, and a classification task branch 53.
The method for training the road extraction model by using the corresponding remote sensing image sample, the binary image sample and the edge symbol distance image sample comprises the following steps:
inputting a remote sensing image sample into a ResNet frame to obtain a remote sensing image characteristic diagram;
inputting the remote sensing image feature map into a distance regression task branch, and calculating a first Loss through a first Loss function between the output result of the distance regression task branch and the edge symbol distance image sample;
and inputting the shallow features and the deep features obtained from the distance regression task branch into a classification task branch, and calculating a second Loss through a second Loss function by using the output result of the classification task branch and the binary image sample.
And determining the final Loss based on the first Loss and the second Loss, and training the road extraction model by using the final Loss.
The road extraction model will be described below with reference to fig. 5.
In this embodiment, a ResNet frame is selected as a backbone network, and considering that a forest boundary road is generally small in width and few in pixels, a model needs to maintain high-frequency details of an input image. Therefore, changing stride of the last two residual blocks of the ResNet frame to 1 can obtain a characteristic map of 1/8 size of the input image. In addition, the expansion convolution is used for expanding the receptive field of the convolution layer, so that the spatial context information is richer. The remote sensing image sample (5 c in the figure) is input into a ResNet frame, and then the obtained remote sensing image feature map is transmitted into a 52 (distance regression task branch).
In the distance regression task branch, a 3 × 3 convolution layer (Conv) is firstly adopted to perform dimension reduction on the remote sensing image feature map generated by the backbone network. Then, a batch processing normalization layer (BN) is adopted to normalize the distribution of each batch of input into standard normal distribution so as to reduce the offset of internal covariates and accelerate the learning process, and finally Tanh is adopted as an activation function to generate a shallow feature map. And then, sequentially carrying out 1 × 1 convolution layer (Conv) and activation function Tanh operation on the shallow feature map to generate a deep feature map, and obtaining a low-resolution distance value prediction result. The Tanh function is used to limit the distance value to the range of [ -1,1], thus keeping consistent with the range of the mark value of the sample (5 a in the figure) obtained by the edge symbol distance transformation. Then, upsampling is performed to obtain an edge symbol distance prediction result (Outputs-a) having the same size as the sample of the remote sensing image (5 c in the figure). The first Loss (Loss-1 in the figure) is calculated using the L1 Loss function with the edge-signed distance image samples as the true values in the distance regression branch.
In the classification task branch (53 in the figure), the shallow feature and the deep feature obtained from the distance regression task branch are directly connected, the classification task branch is input, and the shallow feature graph and the deep feature graph generated in the distance regression task branch are shared between the distance regression task and the two classification tasks, so that the fusion of the distance feature and the semantic feature is realized. The 3 x 3 convolution layer, batch normalization layer, and ReLU activation layer are then input to generate a semantic feature map. And obtaining a classification probability graph through the 1 x 1 convolution layer. And (4) up-sampling the classification probability map to obtain a classification prediction result (Outputs-B) with the same size as the remote sensing image sample. In the branch of the classification task, a binary image sample (5 b in the figure) is used as a true value, and a cross entropy Loss function is adopted to calculate a second Loss (Loss-2 in the figure).
As described above, the model adopts a multi-task training strategy, the regression task branch adopts an L1 Loss function, and the first Loss is:
Figure BDA0002677947010000081
wherein N represents the total number of pixels, BSDgt iRepresents the true value, BSD, of pixel iiIs the predicted edge symbol distance value. The classification task branch adopts a cross entropy Loss function, and the second Loss is as follows:
Figure BDA0002677947010000082
wherein N represents the total number of pixels, yiClass representing true value, piIndicates the category of the predicted value.
The final Loss calculation for the multitask is Lfinal=Lbsd+λ×Lcls
Where λ is a hyperparameter that balances the magnitude difference between the two types of losses and can also be used to control the weight ratio of the classification and regression tasks.
And training the road extraction model by using the final Loss until convergence, and after verifying the model, using the road extraction model for extracting the road.
The distance regression task branch can be regarded as playing an intermediate supervision role in the classification task branch in the whole view.
From the above description, the present document obtains an edge symbol distance image sample by performing edge symbol distance transformation on a binary image of a remote sensing image, and trains a multi-branch road extraction model by using the remote sensing image sample, the binary image sample, and the edge symbol distance image sample. The real-value distance information in the edge symbol distance image can promote the model to better identify the road boundary, and better classify road pixels by learning the distance characteristics to the boundary, rather than learning only spectral characteristics, so as to achieve the purpose of correcting shape deformity or breaking the road.
FIG. 6 is a diagram illustrating predicted results of a road extraction model, according to an example embodiment. Reference numeral 61 denotes a remote sensing image, 62 denotes a true value of a road, 63 denotes a result of prediction using only two classes, and 64 denotes a result of prediction using an edge-symbol distance. Referring to fig. 6, in the road extraction method based on remote sensing images provided herein, a distance regression task is introduced into a road extraction model for supervision, and when a road is extracted, the road continuity performance is improved significantly, so that most of the occlusion problems can be overcome.
Fig. 7 is a block diagram illustrating a road extraction device based on a remote sensing image according to an exemplary embodiment. Referring to fig. 7, the remote sensing image-based road extraction device includes: a binary image making module 701, a sample making module 702, an edge symbol distance transformation module 703, a model training module 704 and a road extraction module 705.
The binary image creation module 701 is configured to draw a road in the remote sensing image and convert the road into a binary image.
The sample preparation module 702 is configured to segment the remote sensing image and the two-class image to prepare a remote sensing image sample and a two-class image sample, wherein the remote sensing image sample and the two-class image sample are in one-to-one correspondence.
The edge symbol distance transform module 703 is configured to perform edge symbol distance transform on the binary image samples, so as to obtain edge symbol distance image samples corresponding to the binary image samples one to one.
The model training module 704 is configured to train a road extraction model using the corresponding remote sensing imagery samples, the two-class image samples, the edge symbol distance image samples.
The road extraction module 705 is configured for road extraction of the remote sensing image using the trained road extraction model.
Fig. 8 is a block diagram illustrating a computer device 800 for remote sensing image-based road extraction according to an exemplary embodiment. For example, the computer device 800 may be provided as a server. Referring to fig. 8, the computer apparatus 800 includes a processor 801, and the number of processors may be set to one or more as necessary. The computer device 800 also includes a memory 802 for storing instructions, such as application programs, that are executable by the processor 801. The number of the memories can be set to one or more according to needs. Which may store one or more application programs. Processor 801 is configured to execute instructions to perform a method of remote sensing imagery based road extraction comprising:
drawing a road in the remote sensing image, and converting the road into a two-classification image;
dividing the remote sensing image and the two classified images to manufacture a remote sensing image sample and two classified image samples, wherein the remote sensing image sample corresponds to the two classified image samples one to one;
performing edge symbol distance conversion on the two classified image samples to obtain edge symbol distance image samples corresponding to the two classified image samples one to one;
training a road extraction model by using the corresponding remote sensing image sample, the two classification image samples and the edge symbol distance image sample;
and carrying out road extraction on the remote sensing image by using the trained road extraction model.
As will be appreciated by one skilled in the art, the embodiments herein may be provided as a method, apparatus (device), or computer program product. Accordingly, the present disclosure may take the form of an entirely hardware embodiment, an entirely software embodiment or an embodiment combining software and hardware aspects. Furthermore, the present disclosure may take the form of a computer program product embodied on one or more computer-usable storage media having computer-usable program code embodied in the medium. Computer storage media includes volatile and nonvolatile, removable and non-removable media implemented in any method or technology for storage of information such as computer readable instructions, data structures, program modules or other data, including, but not limited to, RAM, ROM, EEPROM, flash memory or other memory technology, CD-ROM, Digital Versatile Disks (DVD) or other optical disk storage, magnetic cassettes, magnetic tape, magnetic disk storage or other magnetic storage devices, or any other medium which can be used to store the desired information and which can be accessed by the computer, and the like. In addition, communication media typically embodies computer readable instructions, data structures, program modules or other data in a modulated data signal such as a carrier wave or other transport mechanism and includes any information delivery media as known to those skilled in the art.
The present disclosure is described with reference to flowchart illustrations and/or block diagrams of methods, apparatus (devices) and computer program products according to embodiments herein. It will be understood that each flow and/or block of the flow diagrams and/or block diagrams, and combinations of flows and/or blocks in the flow diagrams and/or block diagrams, can be implemented by computer program instructions. These computer program instructions may be provided to a processor of a general purpose computer, special purpose computer, embedded processor, or other programmable data processing apparatus to produce a machine, such that the instructions, which execute via the processor of the computer or other programmable data processing apparatus, create means for implementing the functions specified in the flowchart flow or flows and/or block diagram block or blocks.
These computer program instructions may also be stored in a computer-readable memory that can direct a computer or other programmable data processing apparatus to function in a particular manner, such that the instructions stored in the computer-readable memory produce an article of manufacture including instruction means which implement the function specified in the flowchart flow or flows and/or block diagram block or blocks
These computer program instructions may also be loaded onto a computer or other programmable data processing apparatus to cause a series of operational steps to be performed on the computer or other programmable apparatus to produce a computer implemented process such that the instructions which execute on the computer or other programmable apparatus provide steps for implementing the functions specified in the flowchart flow or flows and/or block diagram block or blocks.
In this document, the terms "comprises," "comprising," or any other variation thereof, are intended to cover a non-exclusive inclusion, such that an article or apparatus that comprises a list of elements does not include only those elements but may include other elements not expressly listed or inherent to such article or apparatus. Without further limitation, an element defined by the phrase "comprising … …" does not exclude the presence of additional like elements in the article or device comprising the element.
While the preferred embodiments herein have been described, additional variations and modifications of these embodiments may occur to those skilled in the art once they learn of the basic inventive concepts. It is therefore intended that the following appended claims be interpreted as including the preferred embodiments and all such alterations and modifications as fall within the scope of this disclosure.
It will be apparent to those skilled in the art that various changes and modifications may be made herein without departing from the spirit and scope thereof. Thus, it is intended that such changes and modifications be included herein, provided they come within the scope of the appended claims and their equivalents.

Claims (10)

1. The road extraction method based on the remote sensing image is characterized by comprising the following steps:
drawing a road in the remote sensing image, and converting the road into a two-classification image;
dividing the remote sensing image and the two classified images to manufacture a remote sensing image sample and two classified image samples, wherein the remote sensing image sample corresponds to the two classified image samples one to one;
performing edge symbol distance conversion on the two classified image samples to obtain edge symbol distance image samples corresponding to the two classified image samples one to one;
training a road extraction model by using the corresponding remote sensing image sample, the two classification image samples and the edge symbol distance image sample;
and carrying out road extraction on the remote sensing image by using the trained road extraction model.
2. The method for extracting a road based on remote sensing image as claimed in claim 1, wherein said performing edge-symbol distance transformation on said two classified image samples comprises:
determining a distance D from each pixel point in the two classified image samples to a closest point located on a road edgeiBased on DiAnd determining the edge symbol distance of all the pixel points.
3. The method for extracting a road based on remote sensing image as claimed in claim 2,
Figure FDA0002677945000000011
wherein x isiIs the pixel location, xjOn the road edge and xiThe nearest pixel position, ED, is the Euclidean distance.
4. The method of claim 3, wherein the D-based image is used for road extractioniDetermining the total imageThe edge symbol distance of a prime point includes:
edge symbol distance BSDiThen, then
Figure FDA0002677945000000012
Wherein Htanh is a HardTanh function, alpha is a proportionality coefficient, F is a foreground area, and B is a background area.
5. The method for extracting a road based on remote sensing images as claimed in claim 1, wherein the road extraction model comprises a ResNet frame, a distance regression task branch and a classification task branch.
6. The method of claim 5, wherein the training of the road extraction model using the corresponding remote sensing image samples, the two-class image samples, and the edge sign distance image samples comprises:
inputting the remote sensing image sample into a ResNet frame to obtain a remote sensing image characteristic diagram;
inputting the remote sensing image feature map into the distance regression task branch, and calculating a first Loss through a first Loss function between the output result of the distance regression task branch and the edge symbol distance image sample;
and inputting the shallow feature and the deep feature obtained from the distance regression task branch into the classification task branch, and calculating a second Loss through a second Loss function by using the output result of the classification task branch and the binary image sample.
7. The method for extracting a road based on a remote sensing image as claimed in claim 6, further comprising:
and determining a final Loss based on the first Loss and the second Loss, and training the road extraction model by using the final Loss.
8. Road extraction element based on remote sensing image, its characterized in that includes:
the second classification image making module is used for drawing the road in the remote sensing image and converting the road into a second classification image;
the sample manufacturing module is used for segmenting the remote sensing image and the two classified images to manufacture a remote sensing image sample and two classified image samples, and the remote sensing image sample corresponds to the two classified image samples one by one;
the edge symbol distance conversion module is used for carrying out edge symbol distance conversion on the binary image samples to obtain edge symbol distance image samples which correspond to the binary image samples one to one;
the model training module is used for training a road extraction model by using the corresponding remote sensing image sample, the two classification image samples and the edge symbol distance image sample;
and the road extraction module is used for extracting the road of the remote sensing image by using the trained road extraction model.
9. A computer-readable storage medium, on which a computer program is stored, characterized in that the computer program, when executed, implements the steps of the method according to any one of claims 1-7.
10. A computer arrangement comprising a processor, a memory and a computer program stored on the memory, characterized in that the steps of the method according to any of claims 1-7 are implemented when the computer program is executed by the processor.
CN202010953874.4A 2020-09-11 2020-09-11 Road extraction method, device, medium and equipment based on remote sensing image Active CN112070037B (en)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202010953874.4A CN112070037B (en) 2020-09-11 2020-09-11 Road extraction method, device, medium and equipment based on remote sensing image

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202010953874.4A CN112070037B (en) 2020-09-11 2020-09-11 Road extraction method, device, medium and equipment based on remote sensing image

Publications (2)

Publication Number Publication Date
CN112070037A true CN112070037A (en) 2020-12-11
CN112070037B CN112070037B (en) 2022-09-30

Family

ID=73697025

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202010953874.4A Active CN112070037B (en) 2020-09-11 2020-09-11 Road extraction method, device, medium and equipment based on remote sensing image

Country Status (1)

Country Link
CN (1) CN112070037B (en)

Cited By (3)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN113139975A (en) * 2021-04-19 2021-07-20 国交空间信息技术(北京)有限公司 Road characteristic-based road surface segmentation method and device
CN113361371A (en) * 2021-06-02 2021-09-07 北京百度网讯科技有限公司 Road extraction method, device, equipment and storage medium
CN115641512A (en) * 2022-12-26 2023-01-24 成都国星宇航科技股份有限公司 Satellite remote sensing image road identification method, device, equipment and medium

Citations (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN101126812A (en) * 2007-09-27 2008-02-20 武汉大学 High resolution ratio remote-sensing image division and classification and variety detection integration method
CN109800736A (en) * 2019-02-01 2019-05-24 东北大学 A kind of method for extracting roads based on remote sensing image and deep learning

Patent Citations (2)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN101126812A (en) * 2007-09-27 2008-02-20 武汉大学 High resolution ratio remote-sensing image division and classification and variety detection integration method
CN109800736A (en) * 2019-02-01 2019-05-24 东北大学 A kind of method for extracting roads based on remote sensing image and deep learning

Non-Patent Citations (3)

* Cited by examiner, † Cited by third party
Title
NICOLAS AUDEBERT 等: "Distance transform regression for spatially-aware deep semantic segmentation", 《COMPUTER VISION AND IMAGE UNDERSTANDING》 *
TONG WU 等: "Improved Anchor-Free Instance Segmentation for Building Extraction from High-Resolution Remote Sensing Images", 《REMOTE SENSING》 *
李晓慧 等: "结合卷积受限玻尔兹曼机的CV图像分割模型", 《激光与光电子学进展》 *

Cited By (5)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN113139975A (en) * 2021-04-19 2021-07-20 国交空间信息技术(北京)有限公司 Road characteristic-based road surface segmentation method and device
CN113139975B (en) * 2021-04-19 2023-11-17 国交空间信息技术(北京)有限公司 Road feature-based pavement segmentation method and device
CN113361371A (en) * 2021-06-02 2021-09-07 北京百度网讯科技有限公司 Road extraction method, device, equipment and storage medium
CN113361371B (en) * 2021-06-02 2023-09-22 北京百度网讯科技有限公司 Road extraction method, device, equipment and storage medium
CN115641512A (en) * 2022-12-26 2023-01-24 成都国星宇航科技股份有限公司 Satellite remote sensing image road identification method, device, equipment and medium

Also Published As

Publication number Publication date
CN112070037B (en) 2022-09-30

Similar Documents

Publication Publication Date Title
CN112070037B (en) Road extraction method, device, medium and equipment based on remote sensing image
US11651477B2 (en) Generating an image mask for a digital image by utilizing a multi-branch masking pipeline with neural networks
US11393100B2 (en) Automatically generating a trimap segmentation for a digital image by utilizing a trimap generation neural network
CN111738262A (en) Target detection model training method, target detection model training device, target detection model detection device, target detection equipment and storage medium
CN111696110B (en) Scene segmentation method and system
CN116665176B (en) Multi-task network road target detection method for vehicle automatic driving
CN113076871A (en) Fish shoal automatic detection method based on target shielding compensation
CN110852327A (en) Image processing method, image processing device, electronic equipment and storage medium
CN115424017B (en) Building inner and outer contour segmentation method, device and storage medium
CN115205636B (en) Image target detection method, system, equipment and storage medium
CN115631112B (en) Building contour correction method and device based on deep learning
CN115761225A (en) Image annotation method based on neural network interpretability
CN115273032A (en) Traffic sign recognition method, apparatus, device and medium
CN113570540A (en) Image tampering blind evidence obtaining method based on detection-segmentation architecture
CN113158856B (en) Processing method and device for extracting target area in remote sensing image
CN113744280A (en) Image processing method, apparatus, device and medium
CN113628180A (en) Semantic segmentation network-based remote sensing building detection method and system
CN116612280A (en) Vehicle segmentation method, device, computer equipment and computer readable storage medium
CN116524261A (en) Image classification method and product based on multi-mode small sample continuous learning
CN114283431B (en) Text detection method based on differentiable binarization
CN113762266B (en) Target detection method, device, electronic equipment and computer readable medium
CN112446292B (en) 2D image salient object detection method and system
CN113807354B (en) Image semantic segmentation method, device, equipment and storage medium
CN113343817A (en) Unmanned vehicle path detection method and device for target area and medium
CN117423116B (en) Training method of text detection model, text detection method and device

Legal Events

Date Code Title Description
PB01 Publication
PB01 Publication
SE01 Entry into force of request for substantive examination
SE01 Entry into force of request for substantive examination
GR01 Patent grant
GR01 Patent grant