Bilateral attention mechanism-based target detection method under complex background

Abstract

The invention discloses a bilateral attention mechanism-based target detection method under a complex background, which can be used for carrying out accurate foreground target detection under the complex background. The method mainly comprises the following steps: constructing a training set, a verification set and a test set according to a disclosed target detection data set under a complex background; constructing an artificial neural network detection model Bi-SINet based on a bilateral attention mechanism; an SGD optimizer is used on a Pytorch deep learning platform to optimize the Bi-SINet model; and the detection performance of the convergent Bi-SINet network model is evaluated on the constructed test set. Compared with a current main target detection algorithm SINet under a complex background, the method provided by the invention can obtain better detection performance. According to the method, the average absolute error is reduced, a higher enhancement-alignment index, a structure index and a weighted F index are realized, and the method is a more accurate target detection algorithm under a complex background.

Description

technical field [0001] The invention relates to a target detection method under complex background based on bilateral attention mechanism, which is applicable to the technical field of target detection under complex background in computer vision. Background technique [0002] Images and videos are an important source of information for human beings, so the application of computer analysis and processing for massive images and videos has also been vigorously developed. As one of the basic tasks in the field of computer vision, object detection is an important means to help computers understand image data. It has a wide range of application prospects in pedestrian detection, vehicle detection, automatic driving, security systems and medical care. [0003] Thanks to the development of deep learning technology, object detection has attracted extensive attention in recent years and achieved great success, and a large number of efficient detection algorithms have been proposed. E...

AI Technical Summary

Problems solved by technology

[0004] Aiming at the problem of insufficient detection accuracy of existing target detection algorithms in complex backgrounds, the present invention proposes a target detection method in complex backgrounds based on a bilateral attention mechanism based on the current algorithm SINet, which is named Bi- SINet

Images