face_swap.py 文件源码

def get_landmarks(im):
    rects = cascade.detectMultiScale(im, 1.3,5)
    #if len(rects) > 1:
    #    raise TooManyFaces
    if len(rects) == 0:
        raise NoFaces
    print len(rects)
    x,y,w,h =rects[0]
    rect=dlib.rectangle(x,y,x+w,y+h)
    return numpy.matrix([[p.x, p.y] for p in predictor(im, rect).parts()])