Target detection model training method and device, storage medium and computer equipment

Abstract

The invention relates to a target detection model training method and device, a computer readable storage medium and computer equipment, and the method comprises the steps: obtaining a feature map ofa sample image during training, and determining an initial detection box in the feature map according to a preset rotation angle, a preset scale and a preset target aspect ratio; adjusting the position of each initial detection frame to obtain the position information of the prediction detection frame, and adjusting the network parameters of the regression network according to the position information and the real position information in the annotation information of the sample image; predicting the prediction probability of the target corresponding to each preset category according to the target detection area determined by the position information of the prediction detection frame; and adjusting the network parameters of the classification network according to the real category information in the annotation information of the sample image and the prediction probability, and then obtaining a target detection model used for performing target detection on the image. According to the scheme provided by the invention, the target detection model can identify the rotation angle of the target in the image, and the positioned target detection frame is more accurate.

Description

technical field [0001] The present application relates to the field of computer technology, in particular to a training method, device, computer-readable storage medium and computer equipment for a target detection model. Background technique [0002] Target detection, also called target extraction, is an image segmentation technology in the field of computer vision. Target detection can not only segment the target from the image, that is, locate the target position, but also identify the category of the target. [0003] When training the position regression network of the target detection model, the sliding window method is usually used to traverse the regions in the image, and then these regions are screened and checked as candidate rectangular regions for target detection. However, the inventors realized that these candidate rectangles The area is usually a horizontal rectangular area. Such a candidate area can accurately and effectively locate the horizontally placed and...

AI Technical Summary

Problems solved by technology

[0004] Based on this, it is necessary to provide a training method and device for a target detection model in view of the technical problems of inaccuracy and low recognition rate in the way that the existing target detection model locates rotating targets or irregularly shaped targets in the image , computer readable storage media and computer equipment

Images