Human body key point detection model training method and device, detection method and device

Abstract

The invention discloses a human body key point detection model training method and device. The method comprises the following steps: acquiring a training sample data set, wherein the training sample data set comprises a plurality of groups of human body pictures; inputting each group of human body pictures into a neural network model to output first and second predicted human body key points through the neural network model; performing inverse transformation operation on the second predicted human body key points to obtain third predicted human body key points; determining a first loss value according to the human body key points of the original human body picture and the third predicted human body key points, and determining a second loss value according to the original human body picture, the human body key points of the augmented human body picture, the first predicted human body key points and the second predicted human body key points; constructing a loss function based on the first loss value and the second loss value; and updating model parameters of the neural network model according to the loss function until a preset condition is met, and obtaining a trained human body key point detection model. The detection precision of the model can be improved.

Description

technical field [0001] The present application relates to the technical field of image processing, in particular to a human body key point detection model training method, detection method and device. Background technique [0002] Human key point detection is also called human key point positioning, which locates key areas of the human body, including face, hands, and feet. Human body key point detection is an important basic link in human body recognition tasks. Accurate detection of human body key points plays a key role in many scientific research and application topics, such as face posture correction, posture recognition, expression recognition, fatigue monitoring, mouth shape recognition, etc. . [0003] In the prior art, human body key point detection is similar to human body detection. Due to the influence of various factors such as posture, occlusion, and shaking, for example, when detecting human body key points in each video frame in a video, due to the The scre...

AI Technical Summary

Problems solved by technology

[0004] In view of this, a human body key point detection model training method, device, computer equipment and computer readable storage medium are provided to solve the problem of low detection accuracy of existing human body key point detection models

Images