Method, device, and storage medium for key point or joint key point detection and model training

An electronic device, comprising:

• a memory configured to store a program;
• and
• one or more processors coupled to the memory and configured to execute the program stored in the memory to cause the electronic device to perform:
• acquiring a sample image and a first expected heat map of a first size and a second expected heat map of a second size corresponding to key points in the sample image, the first size being smaller than the second size;
• inputting the sample image to a key point detection model to obtain a target predicted heat map corresponding to the key points in the sample image output by the key point detection model;
• wherein the key point detection model is configured to determine a first predicted heat map of the first size and a second predicted heat map of the second size corresponding to the key points in the sample image and configured to correct the second predicted heat map according to the first predicted heat map to obtain the target predicted heat map;
• and
• optimizing the key point detection model according to a difference between the first predicted heat map and the first expected heat map and a difference between the target predicted heat map and the second expected heat map.

A non-transitory computer-readable storage medium storing a set of instructions that are executable by one or more processors of a device to cause the device to perform a method for key point detection, the method comprising:

• acquiring an image to be recognized;
• and
• inputting the image to be recognized into a trained key point detection model to obtain a target heat map output by the key point detection model, the target heat map corresponding to key points in the image to be recognized;
• wherein the key point detection model is configured to:
• determine a first heat map of a first size and a second heat map of a second size corresponding to the key points in the image to be recognized;
• and correct the second heat map according to the first heat map to obtain the target heat map, the first size being smaller than the second size.