Road target detection method and device, electronic equipment and storage medium
A target detection and target technology, applied in the direction of instruments, calculations, character and pattern recognition, etc., can solve the problems of low accuracy and low efficiency of road targets
- Summary
- Abstract
- Description
- Claims
- Application Information
AI Technical Summary
Problems solved by technology
Method used
Image
Examples
Embodiment 1
[0063] figure 1 A schematic diagram of the road target detection process provided by the embodiment of the present invention, the process includes the following steps:
[0064] S101: For each sample image in the training set, input the sample image and corresponding label information into the target detection model; wherein the label information records the coordinate information and category of the real frame of the target.
[0065] The model training method provided by the embodiment of the present invention is applied to an electronic device, and the electronic device may be a device such as a PC or a tablet computer, or may be a server.
[0066] A training set for training the model is pre-stored in the electronic device, and each sample image in the training set has corresponding label information.
[0067] Specifically, a txt file can be used to record the label information. The label information includes the coordinate information and category of the target real frame....
Embodiment 2
[0081] In order to avoid the model overfitting phenomenon caused by too few sample images, on the basis of the above-mentioned embodiments, in the embodiment of the present invention, for each sample image in the training set, the sample image and the corresponding label information Before inputting the target detection model, the method also includes:
[0082] Perform sample enhancement processing on the sample images in the training set to generate new sample images; wherein, the sample enhancement processing includes randomly increasing or reducing the size of the sample images, performing random probability horizontal flipping on the sample images, and performing random probability horizontal flip on the sample images. The brightness is randomly adjusted, the chroma of the sample image is randomly adjusted, and the contrast of the sample image is randomly adjusted.
[0083] In the embodiment of the present invention, the sample images in the training set are enriched by pe...
Embodiment 3
[0089] In the process of training the model, the anchor box needs to be determined in advance, and the target detection model calculates the predicted category and offset of the predetermined anchor box, adjusts the position of the anchor box, and outputs the predicted box of the sample image.
[0090] In the embodiment of the present invention, the process of pre-determining the anchor frame includes:
[0091] The number of anchor boxes is set in advance, and the Kmeans clustering algorithm is used to cluster the real boxes of the sample images in the training set to obtain the anchor boxes of the target detection model, where the distance between the real box and the cluster center box in the clustering process is expressed as d=1-IoU.
[0092] In the embodiment of the present invention, the Kmeans clustering algorithm is used to cluster the real frames of the sample images in the training set, and the number of preset anchor frames is the K value in the Kmeans clustering al...
PUM
Login to View More Abstract
Description
Claims
Application Information
Login to View More 


