Key point detection method and device, electronic equipment and storage medium

Abstract

The invention relates to a key point detection method and device, electronic equipment and a storage medium, and the method comprises the steps: obtaining first feature maps of multiple scales for aninput image, and enabling the scales of the first feature maps to be in a multiple relation; Using a first pyramid neural network to carry out forward processing on each first feature map to obtain second feature maps in one-to-one correspondence with the first feature maps, the second feature maps being the same as the first feature maps in one-to-one correspondence with the second feature maps in scale; carrying out reverse processing on each second feature map by utilizing a second pyramid neural network to obtain third feature maps in one-to-one correspondence with the second feature maps,the scales of the third feature maps being the same as the scales of the second feature maps in one-to-one correspondence with the third feature maps; and performing feature fusion processing on eachthird feature map, and obtaining the position of each key point in the input image by using the feature map after feature fusion processing. The positions of the key points can be accurately extracted.

Description

technical field [0001] The present disclosure relates to the technical field of computer vision, and in particular to a key point detection method and device, electronic equipment and a storage medium. Background technique [0002] Human body key point detection is to detect the position information of key points such as joints or facial features from the human body image, so as to describe the posture of the human body through the position information of these key points. [0003] Because the human body is different in size in the image, the existing technology can usually use the neural network to obtain the multi-scale features of the image, which is used to finally predict the position of the key points of the human body. However, we found that using this method, multi-scale features cannot be fully mined and utilized, and the detection accuracy of key points is low. Contents of the invention [0004] Embodiments of the present disclosure provide a key point detection...

AI Technical Summary

Problems solved by technology

However, we found that using this method, it is not possible to fully mine and utilize multi-scale features, and the detection accuracy of key points is low.

Images