U ‰±Ëhµã@sjddlmZddlmZddlmZddlZddlmZdd„Z ddd „Z dd d „Z ddd„Z dd„Z dS)é)Úabsolute_import)Údivision)Úprint_functionNcCsìt |¡}|d}|d}|d|d tj¡|d<|d tj¡|d<|d tj¡|d<|d tj¡|d<|d tj¡|d<|d jjd d  tj¡d |d <|d jjd d  tj¡d |d <t |d¡d  tj¡|d<|S)a8 load BFM 3DMM model Args: model_path: path to BFM model. Returns: model: (nver = 53215, ntri = 105840). nver: number of vertices. ntri: number of triangles. 'shapeMU': [3*nver, 1] 'shapePC': [3*nver, 199] 'shapeEV': [199, 1] 'expMU': [3*nver, 1] 'expPC': [3*nver, 29] 'expEV': [29, 1] 'texMU': [3*nver, 1] 'texPC': [3*nver, 199] 'texEV': [199, 1] 'tri': [ntri, 3] (start from 1, should sub 1 in python and c++) 'tri_mouth': [114, 3] (start from 1, as a supplement to mouth triangles) 'kpt_ind': [68,] (start from 1) PS: You can change codes according to your own saved data. Just make sure the model has corresponding attributes. Úmodel©rrZshapeMUZexpMUZshapePCZshapeEVZexpEVZexpPCZtriÚC©ÚorderéZ tri_mouthZkpt_ind) ÚsioÚloadmatÚastypeÚnpZfloat32ÚTÚcopyÚint32Zsqueeze)Z model_pathrr©rúV/tmp/pip-unpacked-wheel-5oclok7i/insightface/thirdparty/face3d/morphable_model/load.pyÚload_BFM s ""rú BFM_info.matcCst |¡}|d}|d}|S)a¥ load 3DMM model extra information Args: path: path to BFM info. Returns: model_info: 'symlist': 2 x 26720 'symlist_tri': 2 x 52937 'segbin': 4 x n (0: nose, 1: eye, 2: mouth, 3: cheek) 'segbin_tri': 4 x ntri 'face_contour': 1 x 28 'face_contour_line': 1 x 512 'face_contour_front': 1 x 28 'face_contour_front_line': 1 x 512 'nose_hole': 1 x 142 'nose_hole_right': 1 x 71 'nose_hole_left': 1 x 71 'parallel': 17 x 1 cell 'parallel_face_contour': 28 x 1 cell 'uv_coords': n x 2 Ú model_infor)r r )ÚpathrrrrrÚ load_BFM_info4s rú BFM_UV.matcCst |¡}|djdd}|S)zy load uv coords of BFM Args: path: path to data. Returns: uv_coords: [nver, 2]. range: 0-1 ZUVrr)r r r)rrZ uv_coordsrrrÚload_uv_coordsNs rú pncc_code.matcCst |¡}|dj}|S)a load pncc code of BFM PNCC code: Defined in 'Face Alignment Across Large Poses: A 3D Solution Xiangyu' download at http://www.cbsr.ia.ac.cn/users/xiangyuzhu/projects/3DDFA/main.htm. Args: path: path to data. Returns: pncc_code: [nver, 3] Z vertex_code)r r r)rrZ pncc_coderrrÚload_pncc_codeYs  rc Csp|d t¡}t |ddd…f¡d}td|jddƒD]&}t |t ||dd…f¡d¡}q<| tj¡S)z get nose, eye, mouth index ZsegbinrNr )r ÚboolrZnonzeroÚrangeÚshapeZunion1dr)rZ valid_binZ organ_indÚirrrÚ get_organ_indgs $r!)r)r)r)Ú __future__rrrZnumpyrZscipy.ioÚior rrrrr!rrrrÚs    +