Target detection method and device, electronic equipment and storage medium

Abstract

The invention provides a target detection method and device, electronic equipment and a storage medium, and relates to the field of artificial intelligence. The target detection method comprises the steps of obtaining a to-be-detected image and a pre-trained target detection network; the target detection network comprises a general feature extraction network, a feature pyramid network, a multi-scale balance feature extraction network and a detection head network; performing multi-scale feature extraction on the to-be-detected image based on the general feature extraction network and the feature pyramid network to obtain a multi-scale feature map; performing multi-scale feature fusion processing on the multi-scale feature map based on a multi-scale balance feature extractor to obtain a multi-scale fusion feature map; and performing target detection on the multi-scale fusion feature map through the detection head network, and outputting a target detection result of the to-be-detected image. According to the technical scheme of the embodiment of the invention, the multi-scale feature fusion processing can be performed on the multi-scale feature map through the multi-scale balance feature device, and the detection efficiency and accuracy of the to-be-detected image are improved.

Description

technical field [0001] The present disclosure relates to the field of artificial intelligence, and in particular, to a target detection method, a target detection device, electronic equipment, and a computer-readable storage medium. Background technique [0002] With the deepening of research in the field of autonomous driving, multi-scale detection algorithms are becoming more and more diverse. How to consume less computing resources to perform efficient and accurate multi-scale detection of the road image to be detected is the main technical problem to be solved in the development of the field of autonomous driving. [0003] In the related multi-scale detection algorithm, it is either based on a one-stage network, such as SSD (Single ShotMultiBox Detector, single-shot multi-target detector), YOLO (You Only Look Once, only look once, and target area prediction and target category prediction combined into one), RetinaNet (retina network), or based on a two-stage network, su...

AI Technical Summary

Problems solved by technology

[0006] The purpose of the embodiments of the present disclosure is to provide a target detection method, a target detection device, an electronic device, and a computer-readable storage medium, so as to overcome the low efficiency of multi-scale detection of images and the low accuracy of multi-scale detection at least to a certain extent. low problem

Images