WO2024002014A1 - 交通标线的识别方法、装置、计算机设备和存储介质 - Google Patents
交通标线的识别方法、装置、计算机设备和存储介质 Download PDFInfo
- Publication number
- WO2024002014A1 WO2024002014A1 PCT/CN2023/102444 CN2023102444W WO2024002014A1 WO 2024002014 A1 WO2024002014 A1 WO 2024002014A1 CN 2023102444 W CN2023102444 W CN 2023102444W WO 2024002014 A1 WO2024002014 A1 WO 2024002014A1
- Authority
- WO
- WIPO (PCT)
- Prior art keywords
- target
- probability
- sample
- point cloud
- grid
- 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.)
- Ceased
Links
Classifications
-
- G—PHYSICS
- G06—COMPUTING OR CALCULATING; COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V20/00—Scenes; Scene-specific elements
- G06V20/50—Context or environment of the image
- G06V20/56—Context or environment of the image exterior to a vehicle by using sensors mounted on the vehicle
- G06V20/588—Recognition of the road, e.g. of lane markings; Recognition of the vehicle driving pattern in relation to the road
-
- G—PHYSICS
- G06—COMPUTING OR CALCULATING; COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING OR CALCULATING; COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/764—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using classification, e.g. of video objects
-
- G—PHYSICS
- G06—COMPUTING OR CALCULATING; COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/77—Processing image or video features in feature spaces; using data integration or data reduction, e.g. principal component analysis [PCA] or independent component analysis [ICA] or self-organising maps [SOM]; Blind source separation
- G06V10/774—Generating sets of training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING OR CALCULATING; COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/82—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using neural networks
Definitions
- the present disclosure relates to the field of computer vision technology, and specifically, to a traffic marking recognition method, device, computer equipment and storage medium.
- road images are usually captured by a camera device installed on the vehicle, and then based on the road image, the distance between traffic markings such as lane lines and road boundary lines on the road and the vehicle are determined.
- lane lines and road boundary lines determined based on road images usually have discontinuous and inaccurate line segments.
- Embodiments of the present disclosure provide at least a method, device, computer equipment, and storage medium for identifying traffic markings.
- embodiments of the present disclosure provide a method for identifying traffic markings, including:
- the marking recognition result of the road point cloud data is determined.
- the road point cloud data captured by lidar has morphological stability
- using the acquired road point cloud data to identify traffic marking categories can improve the continuity of the identified traffic markings.
- the probability that each target grid belongs to the traffic marking category is determined. Based on the probability of the traffic marking category to which each target grid belongs, each target grid belonging to the traffic marking category can be accurately determined from multiple target grids. grid, and then use each target raster belonging to the traffic marking category to accurately determine each point cloud point belonging to the traffic marking category in the road point cloud data, and then accurately obtain the traffic markings of each category, that is, the marking recognition result .
- the traffic marking category includes a lane line category and a road boundary line category; the target grid is determined based on the local point cloud data contained in each target grid.
- the probability that the grid belongs to the traffic marking category includes:
- a first probability that the target grid belongs to the lane line category and a second probability that the target grid belongs to the road boundary line category are determined.
- the feature information related to the traffic marking category in each point cloud point can be fully extracted, and then the accurate image of the target grid can be obtained.
- Road feature information Using the road feature information to determine the first probability that the target raster belongs to the lane line category and the second probability that it belongs to the road boundary line category, the target raster can be accurately classified into the corresponding traffic marking category, and then the target raster can be obtained. Accurate traffic marking category.
- the traffic marking category based on each of the target grids respectively belongs to Probability, determine the marking recognition result of the road point cloud data, including:
- a marking line identification result of the road point cloud data is determined.
- the preset threshold can be used to filter out target probabilities with larger probability values.
- the traffic marking category associated with the target probability is more likely to correspond to the traffic marking category to which the target grid belongs. Therefore, based on the target The probabilistically associated traffic marking categories determine the recognition results more accurately. In this way, by using each target probability to determine the recognition results of each target grid, and then determining the marking line recognition results based on each recognition result, the accuracy of the determined marking line recognition results can be improved.
- determining the recognition result of the target grid based on the traffic marking category associated with the target probability includes:
- the target probability includes the first probability and the second probability, determining the maximum probability among the first probability and the second probability;
- the traffic marking category associated with the maximum probability is used as the identification result of the target grid.
- the traffic marking category associated with the maximum probability is used as the identification result of the target grid, which can ensure the uniqueness of the obtained target probability and improve the obtained Identify the reasonableness of the results.
- the road point cloud data is collected by a driving device, and after determining the marking line recognition result of the road point cloud data, the method further includes:
- the traveling device by controlling the driving of the traveling device through the determined marking position and/or marking type, it can ensure that the traveling device travels in a reasonable area and improve the driving safety of the traveling device.
- determining the probability that the target grid belongs to the traffic marking category based on the local point cloud data contained in each target grid includes:
- the probability that the target grid belongs to the traffic marking category is determined based on the local point cloud data contained in each target grid.
- the trained target neural network has reliable prediction accuracy. Using the trained target neural network, the probability that the target grid belongs to the traffic markings of each category can be accurately determined.
- the target neural network is trained according to the following steps:
- the neural network to be trained is iteratively trained until the training cutoff condition is met, and the result is obtained Describe the target neural network.
- the predicted probability includes a first predicted probability that the sample raster belongs to the lane line category, and a second predicted probability that the sample raster belongs to the road boundary line category;
- the neural network to be trained is iteratively trained until the training cutoff condition is met, Obtain the target neural network, including:
- the annotation label information of the sample grid indicates that the sample grid does not belong to the background, based on The first predicted probability, the second predicted probability of the sample grid, the label information of the sample grid, and the label information corresponding to the background label determine the first loss;
- the label information of the sample grid indicates that the sample grid belongs to the background, based on the first predicted probability, the second predicted probability and the label information corresponding to the background label, it is determined second loss;
- the neural network to be trained is iteratively trained until a training cutoff condition is met, and the target neural network is obtained.
- the consistency between the prediction probability of the network output and the annotation label information can be improved, thereby obtaining the target neural network with reliable prediction accuracy. network.
- the first loss includes a first sub-loss and a second sub-loss
- Determining the first loss based on the first prediction probability, the second prediction probability of the sample grid, the label information of the sample grid, and the label information corresponding to the background label includes:
- the second sub-loss is determined based on other prediction probabilities in the first prediction probability and the second prediction probability except the target prediction probability, and annotation label information corresponding to the background label.
- the first sub-loss can represent the difference between the predicted probability output by the network and the annotation label information
- the second sub-loss can represent the difference between the predicted probability output by the network and the annotation label information corresponding to the background label.
- the first sub-loss and the second sub-loss train the network, which can improve the consistency between the prediction probability of the network output and the annotation label information.
- determining the annotation label information of each sample grid includes:
- a top view of the sample is generated; wherein each sample grid corresponds to a pixel in the top view of the sample;
- the annotation label information of the sample grid matching each pixel is determined.
- a top view of the sample is generated based on the local sample point cloud data contained in each sample grid, so that the pixel information of the pixels in the top view of the sample can be used to characterize the local sample point cloud data corresponding to the sample grid; Then use the pixel information of the pixels to determine the label information of the sample grid that matches the pixels, which can reduce the difficulty of labeling and improve the labeling speed.
- determining the annotation label information of the sample grid that matches each pixel based on the pixel information of each pixel in the top view of the sample includes:
- the label information of adjacent pixels of the target pixel is adjusted to the label information of the target pixel.
- the traffic markings of each category have a certain width
- the width and adjusting the label information of adjacent pixels of the target pixel by expanding the width and adjusting the label information of adjacent pixels of the target pixel, the accuracy and accuracy of the label information of adjacent pixels can be improved. rationality.
- an embodiment of the present disclosure also provides a device for identifying traffic markings, including:
- a dividing module used to rasterize the road point cloud data to obtain local point cloud data contained in at least one target grid
- a first determination module configured to determine the local point cloud data contained in each target grid. The probability that the target raster belongs to the traffic marking category;
- the second determination module is used to determine the marking line recognition result of the road point cloud data based on the probability of the traffic marking category to which each of the target grids respectively belongs.
- the traffic marking category includes a lane line category and a road boundary line category;
- the first determination module based on the local points included in each target grid, Cloud data, when determining the probability that the target grid belongs to the traffic marking category, for the local point cloud data contained in each target grid, for each point cloud in the local point cloud data Perform feature extraction on point cloud information of points to generate road feature information of the target grid;
- a first probability that the target grid belongs to the lane line category and a second probability that the target grid belongs to the road boundary line category are determined.
- the second determination module determines the marking recognition result of the road point cloud data based on the probability of the traffic marking category to which each of the target grids respectively belongs. when, for each of the target grids, determine whether there is a target probability greater than a preset threshold in the first probability and the second probability of the target grid;
- a marking line identification result of the road point cloud data is determined.
- the second determination module is configured to determine the recognition result of the target grid based on the traffic marking category associated with the target probability.
- the probability includes the first probability and the second probability, determine the maximum probability among the first probability and the second probability;
- the traffic marking category associated with the maximum probability is used as the identification result of the target grid.
- the road point cloud data is collected by a driving device, and the device further includes:
- a control module configured to control the driving of the driving device based on the marking information of at least one traffic marking indicated by the marking line recognition result after determining the marking line recognition result of the road point cloud data, wherein,
- the reticle information includes reticle position and/or reticle category.
- the first determination module determines the probability that the target grid belongs to the traffic marking category based on the local point cloud data contained in each target grid. is used to use the trained target neural network to determine the probability that the target grid belongs to the traffic marking category based on the local point cloud data contained in each target grid.
- the device further includes:
- the training module is used to train and obtain the target neural network according to the following steps:
- the neural network to be trained is iteratively trained until the training cutoff condition is met, and the result is obtained Describe the target neural network.
- the predicted probability includes a first predicted probability that the sample raster belongs to the lane line category, and a second predicted probability that the sample raster belongs to the road boundary line category;
- the training module performs iterative training on the neural network to be trained based on the predicted probability of the traffic marking category to which each of the sample grids respectively belongs and the annotation label information of each of the sample grids, Until the training cutoff condition is met and the target neural network is obtained, the method is used to, when the annotation label information of the sample grid indicates that the sample grid does not belong to the background, the method based on the sample grid
- the first prediction probability, the second prediction probability, the label information of the sample grid, and the label information corresponding to the background label determine the first loss;
- the label information of the sample grid indicates that the sample grid belongs to the background, based on the first predicted probability, the second predicted probability and the label information corresponding to the background label, it is determined second loss;
- the neural network to be trained is iteratively trained until a training cutoff condition is met, and the target neural network is obtained.
- the first loss includes a first sub-loss and a second sub-loss
- the training module determines the first prediction probability, the second prediction probability based on the sample grid, the label information of the sample grid, and the label information corresponding to the background label. When there is a loss, it is used to determine, from the first predicted probability and the second predicted probability of the sample grid, a target that matches the traffic marking category indicated by the label information of the sample grid. predicted probability;
- the second sub-loss is determined based on other prediction probabilities in the first prediction probability and the second prediction probability except the target prediction probability, and annotation label information corresponding to the background label.
- the training module when determining the annotation label information of each sample grid, is used to based on the local sample point cloud data contained in each of the sample grids, Generate a top view of the sample; wherein each sample grid corresponds to a pixel in the top view of the sample;
- the annotation label information of the sample grid matching each pixel is determined.
- the training module determines the annotation label information of the sample grid that matches each pixel based on the pixel information of each pixel in the top view of the sample. , used to determine the annotation label information corresponding to each pixel in the sample top view based on the pixel information of each pixel in the sample top view;
- the label information of adjacent pixels of the target pixel is adjusted to the label information of the target pixel.
- an optional implementation manner of the present disclosure also provides a computer device, a processor, and a memory.
- the memory stores machine-readable instructions executable by the processor, and the processor is configured to execute the instructions stored in the memory.
- Machine-readable instructions when the machine-readable instructions are executed by the processor, when the machine-readable instructions are executed by the processor, the above-mentioned first aspect, or any possible implementation of the first aspect, is executed. steps in the way.
- an optional implementation manner of the present disclosure also provides a computer-readable storage medium.
- the computer-readable storage medium stores a computer program. When the computer program is run, it executes the above-mentioned first aspect, or any of the first aspects. steps in a possible implementation.
- Figure 1 shows a flow chart of a method for identifying traffic markings provided by an embodiment of the present disclosure
- Figure 2 shows a flow chart of a method for training a neural network to be trained provided by an embodiment of the present disclosure
- Figure 3 shows a schematic diagram of a traffic marking recognition device provided by an embodiment of the present disclosure
- FIG. 4 shows a schematic structural diagram of a computer device provided by an embodiment of the present disclosure.
- the present disclosure provides a traffic marking recognition solution. Since the road point cloud data captured by lidar has morphological stability, using the acquired road point cloud data to identify the traffic marking category can improve the identification. Continuity of traffic markings. By rasterizing the road point cloud data, and then processing the local point cloud data corresponding to each target raster obtained after rasterization, the probability that each target raster belongs to the traffic marking category can be determined.
- each target grid belonging to the traffic marking category can be accurately determined from multiple target grids, and then each target grid belonging to the traffic marking category can be used to accurately determine the road
- Each point cloud point in the point cloud data belongs to the traffic marking category, and then the traffic markings of each category are accurately obtained, that is, the marking recognition result.
- the execution subject of the method for identifying traffic markings provided by an embodiment of the disclosure generally has certain computing capabilities.
- Terminal equipment or other processing equipment where the terminal equipment can be user equipment (User Equipment, UE), mobile equipment, user terminals, terminals, personal digital processing equipment (Personal Digital Assistant, PDA), handheld devices, computer equipment, etc.; in In some possible implementations, the traffic marking recognition method can be implemented by the processor calling computer readable instructions stored in the memory.
- a flow chart of a method for identifying traffic markings may include the following steps:
- the road point cloud data can be obtained using a lidar installed on the driving device.
- the road point cloud data can be a point cloud vector set, and the point cloud vector set includes point cloud information of multiple point cloud points.
- the point cloud information may include the coordinates of the point cloud points in the three-dimensional world coordinate system, the color information of the point cloud points, reflection intensity information, distance information, etc.
- the laser radar installed on the driving device can be used to collect road point cloud data of the driving road.
- S102 Rasterize the road point cloud data to obtain local point cloud data contained in at least one target raster.
- the road point cloud data is rasterized to obtain multiple rasters, wherein the multiple rasters obtained may include empty rasters and non-empty target rasters.
- Each target raster can include local point cloud data.
- the local point cloud data is part of the point cloud data in the road point cloud data, including at least one point cloud point in the road point cloud data, and point cloud information of each point cloud point in the at least one point cloud point.
- the local point cloud data contained in each target raster can form the above-mentioned road point cloud data.
- the horizontal axis direction in the world coordinate system can be the direction of the road
- the vertical axis direction (i.e., the y-axis direction) in the world coordinate system can be perpendicular to the direction of the road.
- the vertical axis direction (i.e., the z-axis direction) can be the direction perpendicular to the road and pointing to the sky (or the ground).
- the road point cloud data can be rasterized and divided according to the preset grid size.
- the preset grid size may be L meters * M meters * N meters, where L is the length in the x-axis direction, M is the length in the y-axis direction, and N is the length in the z-axis direction. length.
- the default grid size can be 0.16m*0.16m*15m.
- the values corresponding to L, M, and N can be determined according to the parameters of the actual lidar, and are not specifically limited in the embodiments of the present disclosure. For example, it can be determined based on the maximum x value, maximum y value, and maximum z value among the coordinates of the point cloud points in the road point cloud data acquired by the lidar.
- the number of divided target grids and the location of each point cloud point can be determined according to the preset grid size and the coordinates of each point cloud point in the road point cloud data in the world coordinate system.
- the target raster uses each point cloud point located in the same target raster as the local point cloud data contained in the target raster.
- At least one target grid can be obtained, and the local point cloud data contained in each target grid in the at least one target grid can be obtained.
- S103 Based on the local point cloud data contained in each target grid, determine the probability that the target grid belongs to the traffic marking category.
- the traffic markings may specifically be any markings on the road.
- traffic markings can be lane lines, road boundary lines, zebra crossings, turn indicators, etc.
- the probability that the target raster belongs to the traffic marking can be determined based on the point cloud information of each point cloud point in the local point cloud data contained in the target raster. . For example, based on the point cloud information of each point cloud point of the target grid, it can be determined whether each point cloud point belongs to the traffic marking line. According to the number of point cloud points belonging to the traffic marking line in the target grid, accounting for The ratio of the total number of point cloud points in the target grid determines the probability that the target grid belongs to the traffic marking category.
- the first number of point cloud points belonging to the traffic markings in the target grid and the first number of point cloud points belonging to the background in the target grid can be determined.
- the second quantity determines the probability that the target grid belongs to the traffic marking category based on the ratio of the first quantity and the second quantity.
- the traffic marking category may include a lane line category and a road boundary line category, that is, the traffic markings may include traffic markings of the lane line category and traffic markings of the road boundary line category.
- the traffic markings of the road boundary line category include road boundary lines located at the edges of both sides of the road; the traffic markings of the lane line category include various traffic markings on the road other than the traffic markings of the non-road boundary line category.
- S103-1 For the local point cloud data contained in each target grid, perform feature extraction on the point cloud information of each point cloud point in the local point cloud data to generate road feature information of the target grid.
- the road feature information is high-dimensional feature information, for example, 64-dimensional, 128-dimensional, etc.
- Road feature information is feature information that can characterize whether the target raster is related to the traffic marking category in the road.
- feature extraction can be performed on the point cloud information of each point cloud point in the local point cloud data contained in the target raster, and the point cloud information of each point cloud point can be extracted.
- target feature information related to the traffic marking category For example, for each point cloud point, the first feature information related to the lane line category can be extracted from the point cloud information of the point cloud point, and, from the point cloud information of the point cloud point, the first feature information related to the lane line category can be extracted. Second feature information related to the road boundary line category. Then, the first feature information and the second feature information can be used as the target feature information of the point cloud point.
- the target feature information corresponding to each point cloud point can be feature fused to obtain the road feature information corresponding to the target grid.
- S103-2 Based on the road feature information, determine the first probability that the target raster belongs to the lane line category and the second probability that the target grid belongs to the road boundary line category.
- the probability that the target grid belongs to the traffic marking category may include a first probability and a second probability.
- the first probability is used to represent the probability that the point cloud points in the target grid belong to the lane line category
- the second probability is used to represent the point cloud points in the target grid that belong to the road boundary line category. The probability.
- the road feature information corresponding to the target grid can be convolved, and based on the results of the convolution process, the first probability that the target grid belongs to the lane line category and the second probability that the target grid belongs to the road boundary line category are determined.
- the above S103 can be executed using the trained target neural network.
- the local point cloud data contained in each target raster can be input into the trained target neural network in a serial manner, using The target neural network processes the local point cloud data contained in each target raster separately, and outputs the first probability that the target raster belongs to the lane line category and the second probability that the target raster belongs to the road boundary line category.
- inputting to the trained target neural network for processing in series can reduce the processing pressure of the target neural network and make the target neural network more lightweight.
- the feature extractor in the target neural network to extract the local point cloud data.
- Feature extraction is performed on the point cloud information of each point cloud point in the data to obtain high-dimensional road feature information of the target raster, and then feature processing is performed on the road feature information to output the first probability sum that the target raster belongs to the lane line category.
- the second probability of belonging to the road boundary line category may include two parts, one part is used to extract road feature information, and the other part is used to perform feature processing on the road feature information, and output the first probability and the second probability.
- the part used to extract road feature information can include a layer of fully connected layer, a layer of batch normalization layer, a layer of linear rectification function (ReLU) function layer and a layer of max pooling layer.
- the local point cloud data contained in the target raster can be input to the fully connected layer, and the point cloud information of each point cloud point in the local point cloud data can be fully connected to obtain the first point cloud point of each point cloud point.
- intermediate feature information then input the first intermediate feature information of each point cloud point into the batch normalization layer, convert the range of feature values corresponding to each first intermediate feature information, and obtain the first intermediate feature information of each point cloud point.
- Second intermediate feature information then input the second intermediate feature information of each point cloud point into the ReLU function layer, use the ReLU function to numerically transform the eigenvalues corresponding to each second intermediate feature information, and set the eigenvalues less than 0 to is 0, retain the feature values greater than 0, thereby obtaining the third intermediate feature information of each point cloud point; finally, the third intermediate feature information of each point cloud point can be input to the maximum pooling layer, using the maximum pooling The layer fuses the third intermediate feature information of each point cloud point to obtain high-dimensional road feature information of the target grid.
- the part of the feature extractor used to output the first probability and the second probability may also include multiple network layers, where the multiple network layers may be a 2-dimensional (2D) convolution layer, a batch normalization layer, and a ReLU.
- Function layer, upsampling layer and sigmoid activation function layer, multiple network layers form a fully convolutional network.
- the sigmoid function is used for the output of hidden layer neurons, and the value range is (0, 1). It can map a value to the interval of (0, 1) and can be used for binary classification.
- the road feature information can be input into the 2D convolution layer, and the road feature information can be 2D convolved to obtain the fourth intermediate feature information; then the fourth intermediate feature information can be input into the batch
- the unified layer converts the range of eigenvalues corresponding to the fourth intermediate feature information to obtain the fifth intermediate feature information; inputs the fifth intermediate feature information to the ReLU function layer, and uses the ReLU function to calculate the features corresponding to the fifth intermediate feature information.
- the value is numerically converted to obtain the sixth intermediate feature information; the sixth intermediate feature information is input to the upsampling layer, and the upsampling layer is used to upsample the sixth intermediate feature information to obtain the seventh intermediate feature information; finally, the seventh intermediate feature information is The intermediate feature information is input to the sigmoid activation function layer, and the sigmoid activation function is used to perform feature processing on the seventh intermediate feature information, and output the first probability map that the target raster belongs to the lane line category and the second probability map that belongs to the road boundary line category. Among them, the probability intervals corresponding to the first probability map and the second probability map are both (0, 1). According to the first probability map, the first probability that the target grid belongs to the lane line category is obtained, and based on the second probability map, the second probability that the target grid belongs to the road boundary line category is obtained.
- the trained target neural network has reliable prediction accuracy.
- the probability that the target grid belongs to each category of traffic markings can be accurately determined.
- the traffic marking recognition method provided by the embodiment of the present disclosure can also be directly executed using the trained target neural network. That is, the trained target neural network can be used to execute the above S101 to S103 and In the following S104, the trained target neural network is used to directly output the reticle recognition result.
- S104 Determine the marking recognition result of the road point cloud data based on the probability of the traffic marking category to which each target grid belongs.
- the marking recognition result is used to indicate the marking information of at least one traffic marking corresponding to the road point cloud data.
- the mark recognition result can indicate the marking information of each lane corresponding to the road point cloud data and/or The marking information of each road boundary line corresponding to the road point cloud data.
- the marking information may include the marking position and/or the marking category of the traffic marking.
- the probability corresponding to the target grid will also only include one, and then it can be obtained from multiple Among the target grids, the marking grids whose probability is greater than the preset probability are screened out.
- each reticle grid with connectivity is determined, and then based on the coordinates of the point cloud points in each reticle grid with connectivity, A traffic marking is determined, and the marking position of the traffic marking can be determined based on the coordinates of the point cloud points in each connected marking grid.
- each traffic marking line corresponding to the road point cloud data can be determined, as well as the marking line information, that is, the marking line recognition result is obtained.
- the probability corresponding to the target grid will also include the probability corresponding to each traffic marking category.
- Probability For example, when the traffic marking category includes the lane line category and the road boundary line category, the probability corresponding to the target grid may include a first probability that the target grid belongs to the lane line category and a second probability that the target grid belongs to the road boundary line category.
- the category probability corresponding to the traffic marking category can be filtered out from the probability corresponding to each target grid. For each target grid, it can be determined whether the category probability corresponding to the target grid is greater than the preset probability. If so, the traffic marking category indicated by the category probability is used as the traffic marking category to which the target grid belongs. If not, it is determined that the target grid does not belong to the traffic marking category. Based on this, the traffic marking category to which each target grid belongs can be determined.
- each target grid with connectivity can be determined based on the position information of each target line grid in the world coordinate system, and then each target grid with connectivity can be determined based on the location information of each target line grid in the world coordinate system.
- the coordinates of the point cloud points in the target grid determine a traffic marking. Based on the coordinates of the point cloud points in each target grid corresponding to the traffic marking, the marking position of the traffic marking is determined; at the same time, the traffic marking can be associated with the class probability of each target grid corresponding to the traffic marking Line category, as the marking category of the traffic marking. In this way, the traffic markings of each category corresponding to the road point cloud data and the marking information of each traffic marking can be obtained.
- the road point cloud data captured by lidar has morphological stability
- using the acquired road point cloud data to identify traffic marking categories can improve the continuity of the identified traffic markings.
- the probability that each target raster belongs to the traffic marking category can be determined.
- each target grid belonging to the traffic marking category can be accurately determined from multiple target grids, and then each target grid belonging to the traffic marking category can be used to accurately determine the road
- Each point cloud point in the point cloud data belongs to the traffic marking category, and then the traffic markings of each category are accurately obtained, that is, the marking recognition result.
- S104 when the traffic marking category includes the lane line category and the road boundary line category, and the probability of the traffic marking category to which the target grid belongs includes the first probability and the second probability, S104 may follow these steps to implement:
- S104-1 For each target grid, determine whether there is a target probability greater than a preset threshold in the first probability and the second probability of the target grid.
- the preset threshold is a preset minimum probability value
- the lane line category and the road boundary line category may correspond to the same preset threshold.
- the preset threshold can be set based on experience and is not specifically limited here.
- the first probability is greater than the preset threshold, it can be determined that the target grid is more likely to belong to the traffic marking category associated with the first probability; similarly, when the second probability is greater than the preset threshold, It can be determined that the target grid is more likely to belong to the traffic marking category of the second probability association.
- the target probability is the probability between the first probability and the second probability that is greater than the preset threshold.
- the first probability and the second probability of the target grid can be compared with a preset threshold respectively, thereby determining whether there is a target probability in the first probability and the second probability. If yes, the following S104-1 may be performed. If not, that is, neither the first probability nor the second probability of the target raster is greater than the preset threshold, it can be determined that the target raster belongs to the background raster, that is, it can be determined that the local point cloud data contained in the target raster Each point cloud point of is a point cloud point of the background category.
- the preset threshold may also include two, specifically a first preset threshold corresponding to the lane line category and a second preset threshold corresponding to the road boundary line category.
- the first probability can be compared with the first preset threshold, and in the first If the probability is greater than the first preset threshold, it is determined that the first probability belongs to the target probability; otherwise, it is determined that the first probability does not belong to the target probability.
- the second probability can be compared with the second preset threshold. If the second probability is greater than the second preset threshold, it is determined that the second probability belongs to the target probability; otherwise, it can be determined that the second probability does not belong to the target probability. .
- the target grid belongs to the background grid, that is, it can be determined that the target grid contains Each point cloud point in the local point cloud data is a point cloud point of the background category.
- S104-2 When there is a target probability, determine the recognition result corresponding to the target grid according to the traffic marking category associated with the target probability.
- the traffic marking category associated with the target probability is the lane line category
- the traffic marking category associated with the target probability is the road boundary. Line category.
- the recognition result is used to indicate the traffic marking category to which the target raster belongs.
- the recognition result corresponding to the target grid can be determined based on the traffic marking category associated with the target probability. For example, when the traffic marking category associated with the target probability is the lane line category, the corresponding recognition result of the target raster is: the target raster belongs to the traffic marking category, that is, the local point cloud data contained in the target raster Each point cloud point in is a point cloud point of the traffic marking category. For another example, when the traffic marking category associated with the target probability is the road boundary line category, the corresponding recognition result of the target raster is: the target raster belongs to the road boundary line category, that is, the local points contained in the target raster Each point cloud point in the cloud data is a point cloud point of the road boundary line category.
- the target probability includes both the first probability and the second probability
- the maximum probability can be screened out from the first probability and the second probability included in the target probability.
- the traffic marking category associated with the maximum probability is used as the recognition result of the target grid. That is, the traffic marking category with the highest probability association is used as the traffic marking category to which the target raster belongs. In this way, the uniqueness of the obtained target probability can be guaranteed and the rationality of the obtained recognition results can be improved.
- S104-3 Based on the recognition results of each target grid, determine the marking line recognition results of the road point cloud data.
- each target grid belonging to the same traffic marking category can be determined based on the recognition results corresponding to each target grid. Then, each target grid with connectivity can be determined based on the coordinates of the point cloud points in the local point cloud data contained in each target grid belonging to the same traffic marking category. According to each target grid with connectivity, The coordinates of the point cloud points in the grid are used to determine a traffic marking with this traffic marking category. Moreover, the marking position of each traffic marking line can be determined based on the coordinates of the point cloud points in the target grid corresponding to each traffic marking line.
- the traffic marking category to which the traffic marking belongs and the marking position of the traffic marking can be used as the marking information of the traffic marking.
- the marking recognition results of the road point cloud data can be determined based on the marking information of each traffic marking.
- the road point cloud data may be collected by the driving device.
- the road point cloud data can be collected using a laser radar installed on the driving device.
- the driving device can also be controlled to drive based on the marking line information of at least one traffic marking line indicated by the marking line recognition result, wherein the marking line information includes the marking line position and/or the marking line. Line category.
- the driving device may include an autonomous vehicle, a manually driven vehicle, a robot, or any other device that can travel on the road.
- the marking position is used to represent the position of the traffic marking in the world coordinate system.
- the marking category is the traffic marking category to which the traffic marking belongs, for example, the lane line category and the road boundary line category.
- the marking information may include but is not limited to the marking position and/or the marking category. For example, it may also include the distance between the traffic marking and the traveling device, the angle between the direction of the traffic marking and the traveling direction of the traveling device, etc.
- each traffic marking indicated by the marking recognition result can be determined, as well as the marking position and marking type of each traffic marking.
- the safe driving area is determined, and the driving device is controlled to drive in the safe driving area.
- an alarm may be prompted. For example, voice alarm prompts, buzzer alarm prompts, etc.
- the embodiment of the present disclosure also provides a method for training the to-be-trained neural network, as shown in Figure 2, which is an implementation of the present disclosure.
- the example provides a flow chart of a method for training a neural network to be trained, which may include the following steps:
- sample point cloud data can be road point cloud data collected using any lidar.
- Sample point cloud data can be a sample point cloud vector set, and the sample point cloud vector set includes point cloud information of multiple sample point cloud points.
- the point cloud information may include the coordinates of the sample point cloud points in the three-dimensional world coordinate system, color information, reflection intensity information, distance information, etc. of the sample point cloud points.
- S202 Rasterize the sample point cloud data to obtain local sample point cloud data contained in at least one sample raster; and determine the annotation label information of each sample raster.
- the annotation label information may specifically be the label value corresponding to the sample raster.
- the annotation label information can be the first label value corresponding to the lane line category; when the sample raster belongs to the road boundary line category, the annotation label information can be the first label value corresponding to the road boundary line category.
- the number of divided sample grids and the size of each sample point cloud can be determined according to the preset grid size and the coordinates of each sample point cloud point in the sample point cloud data in the world coordinate system.
- the sample grid where the point is located takes each sample point cloud point located in the same sample grid as the local sample point cloud data contained in the sample grid.
- the label information corresponding to the sample grid can be determined in advance.
- annotation label information of each sample grid can be determined according to the following steps:
- Step 1 Generate a sample top view based on the local sample point cloud data contained in each sample grid; where each sample grid corresponds to a pixel in the sample top view.
- each sample grid can be projected separately to obtain a top view of the sample.
- the number of pixels in the sample top view is the number of sample grids, and one sample grid corresponds to one pixel in the sample top view.
- the pixel information of the pixel point can be determined based on the point cloud information of each sample point cloud point in the local sample point cloud data contained in the sample grid.
- pixel point 1 corresponds to sample grid 1
- the pixel information of pixel point 1 can be determined based on the point cloud information of each sample point cloud point in the local sample point cloud data contained in sample grid 1.
- Step 2 Based on the pixel information of each pixel in the top view of the sample, determine the annotation label information of the sample grid that matches each pixel.
- the sample grid matching the pixel point is the sample grid corresponding to the pixel point.
- the semantic label of the pixel can be determined based on the pixel information of the pixel.
- semantic labels can include lane line labels, road boundary line labels and background labels.
- the lane line label represents the lane line category
- the road boundary line label represents the road boundary line category
- the background label represents the background category.
- Different semantic labels correspond to different label values. Specifically, the lane line label corresponds to the first label value, the road boundary line label corresponds to the second label value, and the background label corresponds to the third label value.
- the annotation label information of the sample grid that matches each pixel can be determined. For example, for any pixel, the label value corresponding to the semantic label of the pixel can be used as the annotation label information of the sample grid corresponding to the pixel.
- step 2 can be implemented according to the following steps:
- the semantic label of the pixel can be determined based on the pixel information of the pixel in a manual labeling manner, and then the label value corresponding to the semantic label of the pixel can be used as the pixel.
- annotation label information The annotation label information of a pixel is the annotation label information corresponding to the pixel.
- the traffic label category may include a lane line category and a road boundary line category.
- the preset extension width can be determined based on the number of pixels that need to be used and are located on the left and right sides of the target pixel. For example, for any target pixel, if you need one pixel adjacent to the left and right sides of the target pixel (that is, the pixel adjacent to the left of the target pixel, and the adjacent pixel to the right of the target pixel) pixels) to adjust the label information, the default expansion width can be 3 pixels wide.
- each target pixel point belonging to the traffic marking category can be determined based on the annotation label information of each pixel point. Then for each target pixel, the preset extension width can be used to determine two adjacent pixels located on the left and right sides of the target pixel and adjacent to the target pixel. Afterwards, the label information of two adjacent pixels can be adjusted to the label information of the target pixel. Here, if the label information of the adjacent pixels of the target pixel is the same as the label information of the target pixel, If the label information is the same, the label information of adjacent pixels does not need to be adjusted.
- each adjacent pixel point in the top view of the sample with adjusted label information can be used as a sample grid that matches each adjacent pixel point with adjusted label information.
- grid label information At the same time, in the top view of the sample, except for the adjacent pixels whose label information has been adjusted, the unadjusted label information of other pixels can be used as the label information of the sample grid that matches other pixels.
- the other pixels may include unadjusted pixels labeled with label information.
- each lane line and each road boundary line in the sample top view can be determined based on the semantic label of each pixel.
- the corresponding width can be 1 pixel wide.
- the preset expansion width can be used to expand the width of each lane line and each road boundary line. That is, for each pixel in each lane line, the adjacent pixels on the left and right sides of the pixel can be used as added pixels in the lane line, thereby expanding the width of each lane line and obtaining the expansion.
- the annotation label information of the sample grid corresponding to each pixel in the expanded lane line can be set as the first label value corresponding to the lane line category.
- the adjacent pixels on the left and right sides of the pixel can be used as additional pixels in the road boundary line, thereby expanding the width of each road boundary line and obtaining the expansion.
- road boundary line behind the annotation label information of the sample grid corresponding to each pixel in the expanded road boundary line can be set as the second label value corresponding to the road boundary line category.
- the annotation label information of the corresponding sample grid is set to the third label value corresponding to the background category.
- S203 Input the local sample point cloud data contained in each sample grid to the neural network to be trained, and generate a predicted probability that the sample grid belongs to the traffic marking category.
- the neural network to be trained is the target neural network to be trained.
- the predicted probability is the probability that the sample raster output by the neural network to be trained belongs to the traffic marking category.
- the local sample point cloud data contained in each sample raster can be input to the neural network to be trained, and the local sample point cloud data can be processed using the neural network to be trained.
- the output sample raster belongs to the traffic marking. The predicted probability of the category.
- S204 Based on the predicted probability of the traffic marking category to which each sample grid belongs and the annotation label information of each sample grid, iteratively train the neural network to be trained until the training cutoff condition is met, and the target neural network is obtained.
- the training cutoff condition may include that the number of rounds of iterative training reaches a preset number of rounds, and/or the prediction accuracy of the trained neural network reaches a preset accuracy.
- the prediction loss of the neural network to be trained can be determined based on the predicted probability that the sample raster belongs to the traffic marking category and the label information of the sample raster, and the prediction loss is used to iteratively train the neural network to be trained until the training requirements are met. Cutoff conditions to obtain the target neural network.
- the predicted probability may include a first predicted probability that the sample raster belongs to the lane line category, and a second predicted probability that the sample raster belongs to the road boundary line category.
- the first prediction based on the sample grid can be The first loss is determined based on the probability, the second predicted probability, the label information of the sample grid, and the label information corresponding to the background label.
- the first loss may include a first sub-loss and a second sub-loss.
- the step of determining the first loss may be implemented according to the following sub-steps:
- Sub-step 1 Determine the target predicted probability that matches the traffic marking category indicated by the label information of the sample grid from the first predicted probability and the second predicted probability of the sample grid.
- the target prediction probability is the first prediction probability corresponding to the sample grid, and the label value corresponding to the annotation label information is the first label value; in the annotation label
- the target prediction probability is corresponding to the sample raster
- the second prediction probability of , the label value corresponding to the label information is the second label value.
- the traffic marking category to which the sample grid actually belongs can be determined based on the label information of the sample grid. Then, from the first predicted probability and the second predicted probability of the sample grid, a target predicted probability that matches the traffic marking category to which the sample grid actually belongs is determined.
- Sub-step 2 Determine the first sub-loss based on the target prediction probability and the annotation label information of the sample grid.
- the binary cross-entropy function can be used to determine the first sub-loss based on the target prediction probability and the label value corresponding to the annotation label information.
- Sub-step 3 Determine the second sub-loss based on the other predicted probabilities in the first predicted probability and the second predicted probability except the target predicted probability, and the annotation label information corresponding to the background label.
- the background label is the background category
- the annotation label information corresponding to the background label is the third label value.
- the binary cross-entropy function can be used to determine the second sub-loss based on the third label value corresponding to other predicted probabilities and the annotation label information of the background label.
- the target prediction probability is the first prediction probability
- the other prediction probabilities are the second prediction probability.
- the binary cross-entropy function can be used to determine the first sub-loss based on the first prediction probability and the first label value.
- the binary cross-entropy function can be used to determine the second sub-loss based on the second prediction probability and the third label value.
- the target prediction probability is the second prediction probability
- the other prediction probabilities are the first prediction probability.
- the binary cross-entropy function can be used to determine the first sub-loss based on the second prediction probability and the second label value.
- the binary cross-entropy function can be used to determine the second sub-loss based on the first prediction probability and the third label value.
- first sub-loss and second sub-loss can be regarded as the first loss.
- the first prediction probability and the second prediction probability can be used.
- the annotation label information corresponding to the background label determines the second loss.
- the second loss may include a third sub-loss and a fourth sub-loss.
- the binary cross entropy function can be used to determine the third sub-loss based on the first prediction probability and the third label value corresponding to the sample grid; and the binary cross entropy function can be used to determine the third sub-loss based on the second prediction probability corresponding to the sample grid. Predict the probability and the third label value to determine the fourth sub-loss.
- the third sub-loss and the fourth sub-loss as the second loss.
- the neural network to be trained may be iteratively trained until the training cutoff condition is met, and the target neural network is obtained.
- the writing order of each step does not mean a strict execution order and does not constitute any limitation on the implementation process.
- the specific execution order of each step should be based on its function and possible The internal logic is determined.
- the embodiments of the present disclosure also provide a traffic marking identification device corresponding to the traffic marking identification method. Since the problem-solving principle of the device in the embodiments of the present disclosure is the same as that of the traffic markings in the embodiments of the present disclosure, The identification methods are similar, so the implementation of the device can be referred to the implementation of the method, and repeated details will not be repeated.
- a schematic diagram of a traffic marking recognition device provided by an embodiment of the present disclosure includes:
- Acquisition module 301 is used to obtain road point cloud data
- the dividing module 302 is used to rasterize the road point cloud data to obtain local point cloud data contained in at least one target grid;
- the first determination module 303 is configured to determine the probability that the target grid belongs to the traffic marking category based on the local point cloud data contained in each target grid;
- the second determination module 304 is configured to determine the marking line recognition result of the road point cloud data based on the probability of the traffic marking category to which each of the target grids respectively belongs.
- the traffic marking category includes a lane line category and a road boundary line category;
- the first determination module 303 based on the local information contained in each target grid, Point cloud data, when determining the probability that the target grid belongs to the traffic marking category, is used for the local point cloud data contained in each target grid, for each point in the local point cloud data Feature extraction is performed on the point cloud information of the cloud points to generate road feature information of the target grid;
- the second determination module 304 determines the marking identification of the road point cloud data based on the probability of the traffic marking category to which each of the target grids respectively belongs. As a result, for each of the target grids, determine whether there is a target probability greater than a preset threshold in the first probability and the second probability of the target grid;
- a marking line identification result of the road point cloud data is determined.
- the second determination module 304 is configured to determine the recognition result of the target grid according to the traffic marking category associated with the target probability.
- the target probability includes the first probability and the second probability, determine the maximum probability among the first probability and the second probability;
- the traffic marking category associated with the maximum probability is used as the identification result of the target grid.
- the road point cloud data is collected by a driving device, and the device further includes:
- the control module 305 is configured to control the driving of the driving device based on the marking information of at least one traffic marking indicated by the marking line recognition result after the determination of the marking line recognition result of the road point cloud data, wherein , the reticle information includes reticle position and/or reticle category.
- the first determination module 303 determines that the target grid belongs to the traffic marking category based on the local point cloud data contained in each target grid.
- the trained target neural network is used to determine the probability that the target grid belongs to the traffic marking category based on the local point cloud data contained in each target grid.
- the device further includes:
- the training module 306 is used to train and obtain the target neural network according to the following steps:
- the neural network to be trained is iteratively trained until the training cutoff condition is met, and the result is obtained Describe the target neural network.
- the predicted probability includes a first predicted probability that the sample raster belongs to the lane line category, and a second predicted probability that the sample raster belongs to the road boundary line category;
- the training module 306 performs iterative training on the neural network to be trained based on the predicted probability of the traffic marking category to which each of the sample grids respectively belongs and the annotation label information of each of the sample grids. , until the training cutoff condition is met and the target neural network is obtained, for when the annotation label information of the sample grid indicates that the sample grid does not belong to the background, based on all the parameters of the sample grid.
- the first prediction probability, the second prediction probability, the label information of the sample grid, and the label information corresponding to the background label are used to determine the first loss;
- the label information of the sample grid indicates that the sample grid belongs to the background, based on the first predicted probability, the second predicted probability and the label information corresponding to the background label, it is determined second loss;
- the neural network to be trained is iteratively trained until a training cutoff condition is met, and the target neural network is obtained.
- the first loss includes a first sub-loss and a second sub-loss
- the training module 306 determines based on the first predicted probability, the second predicted probability of the sample grid, the label information of the sample grid, and the label information corresponding to the background label.
- the first loss is used to determine, from the first predicted probability and the second predicted probability of the sample grid, the traffic marking category that matches the traffic marking category indicated by the label information of the sample grid. target prediction probability;
- the second sub-loss is determined based on other prediction probabilities in the first prediction probability and the second prediction probability except the target prediction probability, and annotation label information corresponding to the background label.
- the training module 306, when determining the annotation label information of each sample grid is configured to base on the local sample point cloud data contained in each of the sample grids. , generate a top view of the sample; wherein each sample grid corresponds to a pixel in the top view of the sample;
- the annotation label information of the sample grid matching each pixel is determined.
- the training module 306 determines the annotation label information of the sample grid that matches each pixel based on the pixel information of each pixel in the top view of the sample. When, it is used to determine the annotation label information corresponding to each pixel in the sample top view based on the pixel information of each pixel in the sample top view;
- the label information of adjacent pixels of the target pixel is adjusted to the label information of the target pixel.
- a schematic structural diagram of a computer device provided by an embodiment of the present application includes:
- Processor 41 memory 42 and bus 43.
- the memory 42 stores machine-readable instructions executable by the processor 41
- the processor 41 is used to execute the machine-readable instructions stored in the memory 42 .
- the processor 41 executes The following steps: S101: Obtain road point cloud data; S102: Rasterize the road point cloud data to obtain local point cloud data contained in at least one target raster; S103: Based on the data contained in each target raster Based on the local point cloud data, determine the probability that the target grid belongs to the traffic marking category and S104: Based on the probability of the traffic marking category to which each target grid belongs, determine the marking recognition result of the road point cloud data.
- the above-mentioned memory 42 includes a memory 421 and an external memory 422; the memory 421 here is also called an internal memory, and is used to temporarily store the operation data in the processor 41, as well as the data exchanged with external memory 422 such as a hard disk.
- the processor 41 communicates with the processor 42 through the memory 421.
- the external memory 422 performs data exchange.
- the processor 41 and the memory 42 communicate through the bus 43, so that the processor 41 executes the execution instructions mentioned in the above method embodiment.
- Embodiments of the present disclosure also provide a computer-readable storage medium.
- a computer program is stored on the computer-readable storage medium.
- the computer program executes the traffic marking identification method described in the above method embodiment. step.
- the storage medium may be a volatile or non-volatile computer-readable storage medium.
- the computer program product of the traffic marking recognition method provided by the embodiment of the present disclosure includes a computer-readable storage medium storing program code.
- the instructions included in the program code can be used to execute the traffic marking method described in the above method embodiment.
- the computer program product can be implemented specifically through hardware, software or a combination thereof.
- the computer program product is embodied as a computer storage medium.
- the computer program product is embodied as a software product, such as a Software Development Kit (SDK), etc. wait.
- SDK Software Development Kit
- the disclosed devices and methods can be implemented in other ways.
- the device embodiments described above are only illustrative.
- the division of the units is only a logical function division.
- multiple units or components may be combined.
- some features can be ignored, or not implemented.
- the coupling or direct coupling or communication connection between each other shown or discussed may be through some communication interfaces, and the indirect coupling or communication connection of the devices or units may be in electrical, mechanical or other forms.
- the units described as separate components may or may not be physically separated.
- the components shown as units may or may not be physical units, that is, they may be located in one place, or they may be distributed to multiple on the network unit. Some or all of the units can be selected according to actual needs to achieve the purpose of the solution of this embodiment.
- each functional unit in various embodiments of the present disclosure may be integrated into one processing unit, or each unit may exist physically alone, or two or more units may be integrated into one unit.
- the functions are implemented in the form of software functional units and sold or used as independent products, they can be stored in a non-volatile computer-readable storage medium that is executable by a processor.
- the technical solution of the present disclosure is essentially or the part that contributes to the existing technology or the part of the technical solution can be embodied in the form of a software product.
- the computer software product is stored in a storage medium, including Several instructions are used to cause a computer device (which can be a personal computer, a server, or a network device, etc.) to execute all or part of the steps of the methods described in various embodiments of the present disclosure.
- the aforementioned storage media include: U disk, mobile hard disk, read-only memory (ROM), random access memory (Random Access Memory, RAM), magnetic disk or optical disk and other media that can store program code. .
- the products applying the technical solution of this application will clearly inform the personal information processing rules and obtain the individual's independent consent before processing personal information.
- the product applying the technical solution in this application must obtain the individual's separate consent before processing sensitive personal information, and meet the requirement of "express consent" at the same time. For example, setting up clear and conspicuous signs on personal information collection devices such as cameras to inform them that they have entered the scope of personal information collection, and that personal information will be collected.
- personal information processing rules may include personal information processing rules.
- Information processors purposes of personal information processing, processing methods, types of personal information processed, etc.
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- Evolutionary Computation (AREA)
- Computing Systems (AREA)
- Health & Medical Sciences (AREA)
- Multimedia (AREA)
- Artificial Intelligence (AREA)
- Software Systems (AREA)
- General Health & Medical Sciences (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Databases & Information Systems (AREA)
- Medical Informatics (AREA)
- Molecular Biology (AREA)
- General Engineering & Computer Science (AREA)
- Data Mining & Analysis (AREA)
- Mathematical Physics (AREA)
- Computational Linguistics (AREA)
- Biophysics (AREA)
- Biomedical Technology (AREA)
- Life Sciences & Earth Sciences (AREA)
- Traffic Control Systems (AREA)
- Image Analysis (AREA)
Abstract
Description
Claims (14)
- 一种交通标线的识别方法,其特征在于,包括:获取道路点云数据;对所述道路点云数据进行栅格化划分,得到至少一个目标栅格所包含的局部点云数据;基于每个所述目标栅格所包含的所述局部点云数据,确定所述目标栅格属于交通标线类别的概率;基于各个所述目标栅格分别所属的交通标线类别的概率,确定所述道路点云数据的标线识别结果。
- 根据权利要求1所述的方法,其特征在于,所述交通标线类别包括车道线类别和道路边界线类别;所述基于每个所述目标栅格所包含的所述局部点云数据,确定所述目标栅格属于交通标线类别的概率,包括:针对每个所述目标栅格所包含的所述局部点云数据,对所述局部点云数据中每个点云点的点云信息进行特征提取,生成所述目标栅格的道路特征信息;基于所述道路特征信息,确定所述目标栅格属于所述车道线类别的第一概率和属于所述道路边界线类别的第二概率。
- 根据权利要求2所述的方法,其特征在于,所述基于各个所述目标栅格分别所属的交通标线类别的概率,确定所述道路点云数据的标线识别结果,包括:针对每个所述目标栅格,确定所述目标栅格的所述第一概率和所述第二概率中是否存在大于预设阈值的目标概率;在存在所述目标概率的情况下,根据所述目标概率关联的所述交通标线类别,确定所述目标栅格的识别结果;基于各个所述目标栅格的所述识别结果,确定所述道路点云数据的标线识别结果。
- 根据权利要求3所述的方法,其特征在于,所述根据所述目标概率关联的所述交通标线类别,确定所述目标栅格的识别结果,包括:在所述目标概率包括所述第一概率和所述第二概率的情况下,确定所述第一概率和所述第二概率中的最大概率;将所述最大概率关联的所述交通标线类别,作为所述目标栅格的识别结果。
- 根据权利要求1至4任一项所述的方法,其特征在于,所述道路点云数据为行驶装置采集的,在所述确定所述道路点云数据的标线识别结果之后,还包括:基于所述标线识别结果指示的至少一条交通标线的标线信息,控制所述行驶装置行驶,其中,所述标线信息包括标线位置和/或标线类别。
- 根据权利要求1至5任一项所述的方法,其特征在于,所述基于每个所述目标栅格所包含的所述局部点云数据,确定所述目标栅格属于交通标线类别的概率,包括:利用训练后的目标神经网络,基于每个所述目标栅格所包含的所述局部点云数据,确定所述目标栅格属于交通标线类别的概率。
- 根据权利要求6所述的方法,其特征在于,根据下述步骤训练得到所述目标神经网络:获取样本点云数据;对所述样本点云数据进行栅格化划分,得到至少一个样本栅格所包含的局部样本点云数据;并确定每个样本栅格的标注标签信息;将每个所述样本栅格所包含的所述局部样本点云数据输入至待训练神经网络,生成所述样本栅格属于所述交通标线类别的预测概率;基于各个所述样本栅格分别所属的交通标线类别的预测概率、和每个所述样本栅格的标注标签信息,对所述待训练神经网络进行迭代训练,直至满足训练截止条件,得到所述目标神经网络。
- 根据权利要求7所述的方法,其特征在于,所述预测概率包括所述样本栅格属于车道线类别的第一预测概率、和属于道路边界线类别的第二预测概率;所述基于各个所述样本栅格分别所属的交通标线类别的预测概率、和每个所述样本栅格的标注标签信息,对所述待训练神经网络进行迭代训练,直至满足训练截止条件,得到所述目标神经网络,包括:在所述样本栅格的所述标注标签信息指示所述样本栅格不属于背景的情况下,基于所述样本栅格的所述第一预测概率、所述第二预测概率、所述样本栅格的标注标签信息、和背景标签对应的标注标签信息,确定第一损失;在所述样本栅格的所述标注标签信息指示所述样本栅格属于背景的情况下,基于所述第一预测概率、所述第二预测概率和所述背景标签对应的标注标签信息,确定第二损失;基于所述第一损失和所述第二损失中的至少一种,对所述待训练神经网络进行迭代训练,直至满足训练截止条件,得到所述目标神经网络。
- 根据权利要求8所述的方法,其特征在于,所述第一损失包括第一子损失和第 二子损失;所述基于所述样本栅格的所述第一预测概率、所述第二预测概率、所述样本栅格的标注标签信息、和背景标签对应的标注标签信息,确定第一损失,包括:从所述样本栅格的所述第一预测概率和所述第二预测概率中,确定出与所述样本栅格的标注标签信息指示的交通标线类别相匹配的目标预测概率;基于所述目标预测概率和所述样本栅格的标注标签信息,确定第一子损失;基于所述第一预测概率和所述第二预测概率中除所述目标预测概率之外的其他预测概率,和背景标签对应的标注标签信息,确定第二子损失。
- 根据权利要求7至9任一项所述的方法,其特征在于,所述确定每个样本栅格的标注标签信息,包括:基于各个所述样本栅格分别所包含的所述局部样本点云数据,生成样本俯视图;其中,每个所述样本栅格对应所述样本俯视图中的一个像素点;基于所述样本俯视图中各个像素点的像素信息,确定与每个所述像素点匹配的所述样本栅格的标注标签信息。
- 根据权利要求10所述的方法,其特征在于,所述基于所述样本俯视图中各个像素点的像素信息,确定与每个所述像素点匹配的所述样本栅格的标注标签信息,包括:基于所述样本俯视图中各个像素点的像素信息,确定所述样本俯视图中每个像素点对应的标注标签信息;针对所述标注标签信息指示为交通标线类别的目标像素点,基于预设的扩展宽度,将所述目标像素点的相邻像素点的标注标签信息调整为所述目标像素点的标注标签信息;基于所述样本俯视图中所述相邻像素点的调整后的标注标签信息、和除所述相邻像素点之外的其他像素点的未调整的所述标注标签信息,确定与每个所述像素点匹配的所述样本栅格的标注标签信息。
- 一种交通标线的识别装置,其特征在于,包括:获取模块,用于获取道路点云数据;划分模块,用于对所述道路点云数据进行栅格化划分,得到至少一个目标栅格所包含的局部点云数据;第一确定模块,用于基于每个所述目标栅格所包含的所述局部点云数据,确定所述目标栅格属于交通标线类别的概率;第二确定模块,用于基于各个所述目标栅格分别所属的交通标线类别的概率,确定所述道路点云数据的标线识别结果。
- 一种计算机设备,其特征在于,包括:处理器、存储器,所述存储器存储有所述处理器可执行的机器可读指令,所述处理器用于执行所述存储器中存储的机器可读指令,所述机器可读指令被所述处理器执行时,所述处理器执行如权利要求1至11任意一项所述的交通标线的识别方法的步骤。
- 一种计算机可读存储介质,其特征在于,所述计算机可读存储介质上存储有计算机程序,所述计算机程序被计算机设备运行时,所述计算机设备执行如权利要求1至11任意一项所述的交通标线的识别方法的步骤。
Applications Claiming Priority (2)
| Application Number | Priority Date | Filing Date | Title |
|---|---|---|---|
| CN202210773244.8 | 2022-07-01 | ||
| CN202210773244.8A CN115131759B (zh) | 2022-07-01 | 2022-07-01 | 交通标线的识别方法、装置、计算机设备和存储介质 |
Publications (1)
| Publication Number | Publication Date |
|---|---|
| WO2024002014A1 true WO2024002014A1 (zh) | 2024-01-04 |
Family
ID=83381858
Family Applications (1)
| Application Number | Title | Priority Date | Filing Date |
|---|---|---|---|
| PCT/CN2023/102444 Ceased WO2024002014A1 (zh) | 2022-07-01 | 2023-06-26 | 交通标线的识别方法、装置、计算机设备和存储介质 |
Country Status (2)
| Country | Link |
|---|---|
| CN (1) | CN115131759B (zh) |
| WO (1) | WO2024002014A1 (zh) |
Cited By (3)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| CN118999600A (zh) * | 2024-10-24 | 2024-11-22 | 北京集度科技有限公司 | 一种道路标线信息确定方法、车辆、设备及产品 |
| CN119091410A (zh) * | 2024-11-04 | 2024-12-06 | 北京集度科技有限公司 | 一种道路标线检测方法、车辆、设备及产品 |
| CN121545132A (zh) * | 2026-01-21 | 2026-02-17 | 立得空间信息技术股份有限公司 | 基于车载点云图像的道路要素识别方法、装置及存储介质 |
Families Citing this family (2)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| CN115131759B (zh) * | 2022-07-01 | 2025-06-06 | 上海商汤临港智能科技有限公司 | 交通标线的识别方法、装置、计算机设备和存储介质 |
| CN116503383B (zh) * | 2023-06-20 | 2023-09-12 | 上海主线科技有限公司 | 道路曲线检测方法、系统及介质 |
Citations (5)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| CN110363054A (zh) * | 2018-11-16 | 2019-10-22 | 北京京东尚科信息技术有限公司 | 道路标识线识别方法、装置和系统 |
| WO2020103892A1 (zh) * | 2018-11-21 | 2020-05-28 | 北京市商汤科技开发有限公司 | 车道线检测方法、装置、电子设备及可读存储介质 |
| CN113298910A (zh) * | 2021-05-14 | 2021-08-24 | 阿波罗智能技术(北京)有限公司 | 生成交通标志线地图的方法、设备和存储介质 |
| CN113971221A (zh) * | 2020-07-22 | 2022-01-25 | 上海商汤临港智能科技有限公司 | 一种点云数据的处理方法、装置、电子设备及存储介质 |
| CN115131759A (zh) * | 2022-07-01 | 2022-09-30 | 上海商汤临港智能科技有限公司 | 交通标线的识别方法、装置、计算机设备和存储介质 |
Family Cites Families (8)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| CN108216229B (zh) * | 2017-09-08 | 2020-01-10 | 北京市商汤科技开发有限公司 | 交通工具、道路线检测和驾驶控制方法及装置 |
| US10657390B2 (en) * | 2017-11-27 | 2020-05-19 | Tusimple, Inc. | System and method for large-scale lane marking detection using multimodal sensor data |
| KR102633140B1 (ko) * | 2018-10-23 | 2024-02-05 | 삼성전자주식회사 | 주행 정보를 결정하는 방법 및 장치 |
| CN111209780A (zh) * | 2018-11-21 | 2020-05-29 | 北京市商汤科技开发有限公司 | 车道线属性检测方法、装置、电子设备及可读存储介质 |
| CN111353512B (zh) * | 2018-12-20 | 2023-07-28 | 长沙智能驾驶研究院有限公司 | 障碍物分类方法、装置、存储介质和计算机设备 |
| KR20210102182A (ko) * | 2020-02-07 | 2021-08-19 | 선전 센스타임 테크놀로지 컴퍼니 리미티드 | 도로 표시 인식 방법, 지도 생성 방법, 및 관련 제품 |
| CN113970758A (zh) * | 2020-07-22 | 2022-01-25 | 上海商汤临港智能科技有限公司 | 点云数据处理的方法、及装置 |
| CN113191261B (zh) * | 2021-04-29 | 2022-12-06 | 北京百度网讯科技有限公司 | 图像类别的识别方法、装置和电子设备 |
-
2022
- 2022-07-01 CN CN202210773244.8A patent/CN115131759B/zh active Active
-
2023
- 2023-06-26 WO PCT/CN2023/102444 patent/WO2024002014A1/zh not_active Ceased
Patent Citations (5)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| CN110363054A (zh) * | 2018-11-16 | 2019-10-22 | 北京京东尚科信息技术有限公司 | 道路标识线识别方法、装置和系统 |
| WO2020103892A1 (zh) * | 2018-11-21 | 2020-05-28 | 北京市商汤科技开发有限公司 | 车道线检测方法、装置、电子设备及可读存储介质 |
| CN113971221A (zh) * | 2020-07-22 | 2022-01-25 | 上海商汤临港智能科技有限公司 | 一种点云数据的处理方法、装置、电子设备及存储介质 |
| CN113298910A (zh) * | 2021-05-14 | 2021-08-24 | 阿波罗智能技术(北京)有限公司 | 生成交通标志线地图的方法、设备和存储介质 |
| CN115131759A (zh) * | 2022-07-01 | 2022-09-30 | 上海商汤临港智能科技有限公司 | 交通标线的识别方法、装置、计算机设备和存储介质 |
Cited By (3)
| Publication number | Priority date | Publication date | Assignee | Title |
|---|---|---|---|---|
| CN118999600A (zh) * | 2024-10-24 | 2024-11-22 | 北京集度科技有限公司 | 一种道路标线信息确定方法、车辆、设备及产品 |
| CN119091410A (zh) * | 2024-11-04 | 2024-12-06 | 北京集度科技有限公司 | 一种道路标线检测方法、车辆、设备及产品 |
| CN121545132A (zh) * | 2026-01-21 | 2026-02-17 | 立得空间信息技术股份有限公司 | 基于车载点云图像的道路要素识别方法、装置及存储介质 |
Also Published As
| Publication number | Publication date |
|---|---|
| CN115131759B (zh) | 2025-06-06 |
| CN115131759A (zh) | 2022-09-30 |
Similar Documents
| Publication | Publication Date | Title |
|---|---|---|
| Min et al. | Traffic sign recognition based on semantic scene understanding and structural traffic sign location | |
| US12585944B2 (en) | Data processing system, object detection method, and apparatus thereof | |
| CN110148196B (zh) | 一种图像处理方法、装置以及相关设备 | |
| Hoang et al. | Enhanced detection and recognition of road markings based on adaptive region of interest and deep learning | |
| CN111709923B (zh) | 一种三维物体检测方法、装置、计算机设备和存储介质 | |
| US11410549B2 (en) | Method, device, readable medium and electronic device for identifying traffic light signal | |
| CN115131759B (zh) | 交通标线的识别方法、装置、计算机设备和存储介质 | |
| CN108280397B (zh) | 基于深度卷积神经网络的人体图像头发检测方法 | |
| CN113674287A (zh) | 高精地图的绘制方法、装置、设备以及存储介质 | |
| CN108564088A (zh) | 车牌识别方法、装置、设备及可读存储介质 | |
| CN101859382A (zh) | 一种基于最大稳定极值区域的车牌检测与识别的方法 | |
| CN111259796A (zh) | 一种基于图像几何特征的车道线检测方法 | |
| CN112819753B (zh) | 一种建筑物变化检测方法、装置、智能终端及存储介质 | |
| CN115631344A (zh) | 一种基于特征自适应聚合的目标检测方法 | |
| Mijić et al. | Traffic sign detection using YOLOv3 | |
| CN115063765B (zh) | 一种道路边界线确定方法、装置、设备及存储介质 | |
| CN114399768A (zh) | 基于Tesseract-OCR引擎的工件产品序列号识别方法、装置及系统 | |
| CN116129386A (zh) | 可行驶区域检测方法、系统及计算机可读介质 | |
| CN113449629B (zh) | 基于行车视频的车道线虚实识别装置、方法、设备及介质 | |
| Qiu et al. | Lightweight cross-modal information measure and propagation for road extraction from remote sensing image and trajectory/LiDAR | |
| CN115760878B (zh) | 三维图像实体分割方法、装置、设备、存储介质及车辆 | |
| Daraghmi et al. | Accurate real-time traffic sign recognition based on the connected component labeling and the color histogram algorithms | |
| CN106529391B (zh) | 一种鲁棒的限速交通标志检测与识别方法 | |
| CN114565902A (zh) | 基于联通区域提取和关键点拟合的车道检测方法及系统 | |
| CN119380311B (zh) | 一种交通路牌目标检测识别方法和系统 |
Legal Events
| Date | Code | Title | Description |
|---|---|---|---|
| 121 | Ep: the epo has been informed by wipo that ep was designated in this application |
Ref document number: 23830191 Country of ref document: EP Kind code of ref document: A1 |
|
| NENP | Non-entry into the national phase |
Ref country code: DE |
|
| 122 | Ep: pct application non-entry in european phase |
Ref document number: 23830191 Country of ref document: EP Kind code of ref document: A1 |
|
| 32PN | Ep: public notification in the ep bulletin as address of the adressee cannot be established |
Free format text: NOTING OF LOSS OF RIGHTS PURSUANT TO RULE 112(1) EPC (EPO FORM 1205A DATED 11/06/2025) |
|
| 122 | Ep: pct application non-entry in european phase |
Ref document number: 23830191 Country of ref document: EP Kind code of ref document: A1 |