Key point detection model training method and device, equipment and storage medium

• Main Authors: NIU ZHIBO, LAI BAOHUA, LIU QIWEN, CHEN JIANYE, WANG GUANZHONG, YU DIANHAI, DANG QINGQING, MA YANJUN, DENG KAIPENG
• Format: Patent
• Language: Chinese; English
• Published: 15.03.2022
• Subjects: CALCULATING; COMPUTER SYSTEMS BASED ON SPECIFIC COMPUTATIONAL MODELS; COMPUTING; COUNTING; HANDLING RECORD CARRIERS; IMAGE DATA PROCESSING OR GENERATION, IN GENERAL; PHYSICS; PRESENTATION OF DATA; RECOGNITION OF DATA; RECORD CARRIERS

The invention provides a key point detection model training method and device, equipment and a storage medium, relates to the field of artificial intelligence, in particular to a computer vision and deep learning technology, and can be applied to a key point detection scene of a mobile terminal or an edge terminal. According to one specific embodiment, the method comprises the steps that a training sample set is obtained, training samples comprise sample images and heat maps corresponding to key points in the sample images, and the heat maps corresponding to the key points are generated through the following steps that the quotient of the size of the heat maps and the size of the sample images is determined, determining the product of the obtained quotient and the coordinate of the key point in the sample image as the coordinate of the key point in the heat map, and generating the heat map of the key point according to the determined coordinate; and taking a sample image in the training sample set as input of