beautifierCNN.py 文件源码

def decomposePose(mesh, pose, im):
    modelview = np.matrix(pose.get_modelview())
    proj = np.matrix(pose.get_projection())
    viewport = np.array([0, im.shape[0], im.shape[1], -im.shape[0]])

    modelview = modelview.tolist()
    projection = proj.tolist()
    viewport = viewport.tolist()

    ALL_FACE_MESH_VERTS = BFM_FACEFITTING.landmarks_2_vert_indices[ALL_FACE_LANDMARKS]
    ALL_FACE_MESH_VERTS = np.delete(ALL_FACE_MESH_VERTS, np.where(ALL_FACE_MESH_VERTS == -1)).tolist()
    verts2d = projectMeshTo2D(mesh, pose, im)
    convexHullIndexs = cv2.convexHull(verts2d.astype(np.float32), returnPoints=False)
    warpPointIndexs = convexHullIndexs.flatten().tolist() + ALL_FACE_MESH_VERTS
    indexs = warpPointIndexs

    return modelview, projection, viewport, indexs