Target detection method and device and corresponding model training method and device

Abstract

The invention discloses a target detection method and device and a corresponding model training method and device. The target detection model training method comprises the following steps: analyzing original training data with labels, and determining label category distribution contained in the original training data; performing data augmentation on the original training data according to the annotation category distribution to obtain training data; and performing iterative training on a target detection model according to the training data until a training stop condition is met. The method has the beneficial effects that data augmentation can be carried out on few sample categories in the original training data according to the distribution condition of the original training data categories; and iterative training is carried out according to the augmented original data, so that the long-tail effect of a detection result caused by imbalance of the original data is relieved, the detection accuracy of multiple categories is improved, and large-category target detection is realized.

Description

technical field [0001] The present application relates to the field of target detection, in particular to a target detection method and device and a corresponding model training method and device. Background technique [0002] In the existing automatic driving scene, it is a common method to use lidar to detect the target in the scene, determine the relative position between the target and the automatic driving device, the type of the target and other information. But the target category that existing technology can detect is less, for example, can only judge that a target is a car, but can't identify specifically whether it is a large car or a medium-sized car, whether it is a van or an SUV (sport utility vehicle), that is, only Scenes that can support small-category recognition cannot effectively identify large-category targets. Contents of the invention [0003] In view of the above problems, the present application is proposed to provide an object detection method and...

AI Technical Summary

Problems solved by technology

But the target category that existing technology can detect is less, for example, can only judge that a target is a car, but can't identify specifically whether it is a large car or a medium-sized car, whether it is a van or an SUV (sport utility vehicle), that is, only A scene that can support small category recognition, but cannot effectively identify large category targets

Images