Fast Object Detection Method Based on Lightweight Neural Network

Abstract

The invention discloses a fast object detection method based on a lightweight neural network. Including building convolutional neural network, training convolutional neural network and target detection: defining the main modules of convolutional neural network Front module and Tinier module; defining the number of layers of convolutional neural network and the position of pooling layer; preprocessing the data set, Convert it into the standard format of the target detection framework Darknet input; initialize the parameters of the convolutional neural network; train the neural network model through continuous iterative forward propagation and back propagation; input the test image, and use the neural network model obtained from the training process to calculate Obtain the detection numerical results; mark the image according to the detection results, and mark the position and category of each object with a rectangular frame. The detection speed using the method of the invention is faster and the accuracy rate is also higher.

Description

technical field [0001] The invention relates to the fields of pattern recognition and video analysis, more specifically, a fast target detection method based on a lightweight neural network. Background technique [0002] Object detection is an important research topic in the field of computer vision. It has been widely used in multiple real-world applications, such as face recognition, traffic safety, crowd monitoring and image retrieval. Real-time target detection based on deep learning refers to marking the position and category of target objects in a natural scene picture or video. In the face of massive image and video data, manual labeling is time-consuming and inefficient, and automatic and fast object detection methods are urgently needed. [0003] The object detection method based on deep learning contains two key steps: feature extraction and classifier (regressor) training. Unlike the traditional method of manually extracting the features of the target object, d...

AI Technical Summary

Problems solved by technology

[0004] The present invention aims at the problems of time-consuming and inefficient manual extraction of image features in the above-mentioned target detection method, the large volume of the neural network in the deep learning method consumes storage space, the complexity of the network model, and the slow detection speed, and proposes a lightweight neural network-based Fast Object Detection Method

Images