File size: 1,286 Bytes
ddadf19 f3b7b20 ddadf19 |
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 41 |
# coding: utf-8
__author__ = 'cleardusk'
import sys
sys.path.append('..')
import os.path as osp
import numpy as np
from tddfa.utils.io import _load
make_abs_path = lambda fn: osp.join(osp.dirname(osp.realpath(__file__)), fn)
def _to_ctype(arr):
if not arr.flags.c_contiguous:
return arr.copy(order='C')
return arr
class BFMModel(object):
def __init__(self, bfm_fp, shape_dim=40, exp_dim=10):
bfm = _load(bfm_fp)
self.u = bfm.get('u').astype(np.float32) # fix bug
self.w_shp = bfm.get('w_shp').astype(np.float32)[..., :shape_dim]
self.w_exp = bfm.get('w_exp').astype(np.float32)[..., :exp_dim]
if osp.split(bfm_fp)[-1] == 'bfm_noneck_v3.pkl':
self.tri = _load(make_abs_path('../configs/tri.pkl')) # this tri/face is re-built for bfm_noneck_v3
else:
self.tri = bfm.get('tri')
self.tri = _to_ctype(self.tri.T).astype(np.int32)
self.keypoints = bfm.get('keypoints').astype(np.int64) # fix bug
w = np.concatenate((self.w_shp, self.w_exp), axis=1)
self.w_norm = np.linalg.norm(w, axis=0)
self.u_base = self.u[self.keypoints].reshape(-1, 1)
self.w_shp_base = self.w_shp[self.keypoints]
self.w_exp_base = self.w_exp[self.keypoints]
|