US 12,080,098 B2
Method and device for training multi-task recognition model and computer-readable storage medium
Yusheng Zeng, Shenzhen (CN); Jun Cheng, Shenzhen (CN); and Jianxin Pang, Shenzhen (CN)
Assigned to UBTECH ROBOTICS CORP LTD, Shenzhen (CN)
Filed by UBTECH ROBOTICS CORP LTD, Shenzhen (CN)
Filed on Dec. 27, 2021, as Appl. No. 17/562,963.
Application 17/562,963 is a continuation of application No. PCT/CN2020/139615, filed on Dec. 25, 2020.
Prior Publication US 2022/0207913 A1, Jun. 30, 2022
Int. Cl. G06V 40/16 (2022.01); G06T 7/73 (2017.01)
CPC G06V 40/171 (2022.01) [G06T 7/73 (2017.01); G06V 40/166 (2022.01)] 20 Claims
OG exemplary drawing
 
1. A computer-implemented method for training a multi-task recognition model, comprising:
obtaining a first set of sample images, a second set of sample images, and a third set of sample images, wherein the first set of sample images comprise a plurality of sample images that are configured to provide feature-independent facial attributes, the second set of sample images comprise a plurality of sample images that are configured to provide feature-coupled facial attributes, and the third set of sample images comprise a plurality of sample images that are configured to provide facial attributes of face poses;
training an initial feature-sharing model based on the first set of sample images to obtain a first feature-sharing model with a loss value less than a preset first threshold;
training the first feature-sharing model based on the first set of sample images and the second set of sample images to obtain a second feature-sharing model with a loss value less than a preset second threshold;
obtaining an initial multi-task recognition model by adding a feature decoupling model to the second feature-sharing model; and
training the initial multi-task recognition model based on the first set of sample images, the second set of sample images, and the third set of sample images to obtain a trained multi-task recognition model with a loss value less than a preset third threshold.