| import numpy as np | |
| from scipy.io import loadmat | |
| original_BFM = loadmat("3DMM/01_MorphableModel.mat") | |
| sub_inds = np.load("3DMM/topology_info.npy", allow_pickle=True).item()["sub_inds"] | |
| shapePC = original_BFM["shapePC"] | |
| shapeEV = original_BFM["shapeEV"] | |
| shapeMU = original_BFM["shapeMU"] | |
| texPC = original_BFM["texPC"] | |
| texEV = original_BFM["texEV"] | |
| texMU = original_BFM["texMU"] | |
| b_shape = shapePC.reshape(-1, 199).transpose(1, 0).reshape(199, -1, 3) | |
| mu_shape = shapeMU.reshape(-1, 3) | |
| b_tex = texPC.reshape(-1, 199).transpose(1, 0).reshape(199, -1, 3) | |
| mu_tex = texMU.reshape(-1, 3) | |
| b_shape = b_shape[:, sub_inds, :].reshape(199, -1) | |
| mu_shape = mu_shape[sub_inds, :].reshape(-1) | |
| b_tex = b_tex[:, sub_inds, :].reshape(199, -1) | |
| mu_tex = mu_tex[sub_inds, :].reshape(-1) | |
| exp_info = np.load("3DMM/exp_info.npy", allow_pickle=True).item() | |
| np.save( | |
| "3DMM/3DMM_info.npy", | |
| { | |
| "mu_shape": mu_shape, | |
| "b_shape": b_shape, | |
| "sig_shape": shapeEV.reshape(-1), | |
| "mu_exp": exp_info["mu_exp"], | |
| "b_exp": exp_info["base_exp"], | |
| "sig_exp": exp_info["sig_exp"], | |
| "mu_tex": mu_tex, | |
| "b_tex": b_tex, | |
| "sig_tex": texEV.reshape(-1), | |
| }, | |
| ) | |