METHOD AND DEVICE FOR TRAINING MULTI-TASK RECOGNITION MODEL AND COMPUTER-READABLE STORAGE MEDIUM
Abstract:
A method for training a multi-task recognition model includes: obtaining a number of sample images, wherein some of the sample images are to provide feature-independent facial attributes, some of the sample images are to provide feature-coupled facial attributes, and some of the sample images are to provide facial attributes of face poses; training an initial feature-sharing model based on a first set of sample images to obtain a first feature-sharing model; training the first feature-sharing model based on the first set of sample images and a 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 sample images to obtain a trained multi-task recognition model.
Information query
Patent Agency Ranking
0/0