Spaces:
Sleeping
Sleeping
File size: 1,189 Bytes
bc3753a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
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),
},
)
|