Target detection method based on DenseNet and multi-scale feature fusion

Abstract

The invention provides a target detection method based on DenseNet and multi-scale feature fusion. The target detection method comprises the following steps: S1, constructing a feature extraction network model; S2, training the feature extraction network model, and obtaining an optimal target detection model through multiple times of iterative training; S3, inputting the to-be-detected image datainto the optimal target detection model for detection, and marking the position and the category of each object on the to-be-detected image by using a rectangular frame. According to the feature extraction network model, a DenseNet network serves as a basic network, the network hierarchy is deepened, the feature quality is improved, meanwhile, a feature fusion module is used, context information is introduced, six feature maps used for final prediction are obtained, and the feature extraction network model has rich semantic information and high resolution. According to the method, the model scale can be reduced on the basis of ensuring the detection speed, and the detection precision of a small target is improved.

Description

technical field [0001] The invention relates to the technical field of computer vision, in particular to a target detection method based on DenseNet and multi-scale feature fusion. Background technique [0002] Object detection is a very important research direction in the field of computer vision. Target detection is to accurately identify and locate objects in images and videos, so that the computer can understand the surrounding environment and achieve good human-computer interaction. In recent years, object detection has been widely used in autonomous driving, environmental monitoring, traffic security and other fields. [0003] In practical applications, there are many scenes that need to identify small targets in the image to be detected. However, because the resolution and information of the small target object in the image to be detected is limited, so far, the detection of the small target is still a difficult point in the prior art. Currently, object detection m...

AI Technical Summary

Problems solved by technology

[0004] In order to solve the problem that the detection accuracy and detection speed cannot be guaranteed at the same time for small target detection in the prior art, the present invention provides a target detection method based on DenseNet and multi-scale feature fusion, which can reduce the model scale on the basis of ensuring the detection speed, Improve the accuracy of small target detection

Images