Target detection model training method and device and electronic equipment

Abstract

The invention provides a target detection model training method and device and electronic equipment. The method comprises the following steps: acquiring reference information of a reference frame in atraining image; obtaining prediction information obtained through analysis of the target detection model; wherein the prediction information comprises prediction center point information and prediction size information corresponding to all sample points in the center area, the center area is a sub-area of the reference frame, the center point of the center area is an object center point, and theprediction size information comprises sample point information and corresponding distance information; and obtaining and repeatedly training the target detection model according to the first evaluation result and the second evaluation result until the target detection model meets the expected requirements. The second evaluation result comprises the size data corresponding to all the sample points,the data size evaluated by the second evaluation result is large and rapid to obtain, the convergence rate of the target verification model is increased, the training frequency of the target detection model is reduced, and the training speed is increased.

Description

Technical field [0001] The invention relates to the field of Internet technology, and in particular to a training method, device and electronic equipment of a target detection model. Background technique [0002] Target detection is a research field that promotes and develops with deep learning. The image data obtained by camera sensors can be applied to real-life scenes such as vehicle detection, pedestrian detection, traffic light detection, and many fields such as unmanned driving and security systems. [0003] Existing target detection methods use target detection models to locate and recognize target objects. Most target detection models are based on anchor designs. Recently, the anchor-free design method has emerged. The training process of the target detection model based on anchor-free design is to input the original image and label information (including the reference frame and the category corresponding to the target object) of the training target verification model into ...

AI Technical Summary

Problems solved by technology

[0005] The present invention provides a training method, device and electronic equipment of a target detection model to solve the problem of long training time of the existing target detection model

Images