Method and apparatus for training parameter estimation models, device, and storage medium

Abstract

Provided is a method for training parameter estimation models. The method includes: estimating reconstruction parameters specified for three-dimensional face reconstruction by inputting training samples in a face image training set to a pre-constructed neural network model, and reconstructing three-dimensional faces corresponding to the training samples by inputting the reconstruction parameters to a pre-constructed three-dimensional morphable model; calculating a plurality of loss functions of a plurality of pieces of two-dimensional supervision information between the three-dimensional faces and the training samples, and adjusting weights corresponding to the plurality of loss functions; and generating fitting loss functions based on the plurality of loss functions and the weights corresponding to the plurality of loss functions, and acquiring a trained parameter estimation model by performing an inverse correction on the neural network model using the fitting loss functions.