test_alignment.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# -----------------------------------------------
# 测试人脸对齐
# python ./tests/test_alignment.py
# -----------------------------------------------

# 把父级目录（项目根目录）添加到工作路径
import os
import sys
sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/../')
# ----------------------------------------------------------------------

import argparse
from src.utils.common import input_face
from src.cache.face_feature_cache import FACE_FEATURE_CACHE
from src.core.face_detection import FaceDetection
from src.core.face_alignment import FaceAlignment


def args() :
    parser = argparse.ArgumentParser(
        formatter_class=argparse.RawDescriptionHelpFormatter,
        prog='测试人脸对齐',
        description='测试人脸对齐（仿射变换）', 
        epilog='\r\n'.join([
            '摄像头拍摄: ', 
            '  python ./tests/test_alignment.py -c', 
            '上传人脸图片: ', 
            '  python ./tests/test_alignment.py', 
            '注意：', 
            '  需要至少执行一次 python ./presrc/gen_standard.py 生成标准脸数据，作为仿射变换参照系', 
        ])
    )
    parser.add_argument('-c', '--camera', dest='camera', action='store_true', default=False, help='摄像头模式; 默认为图片上传模式')
    return parser.parse_args()


def test(args) :
    FACE_FEATURE_CACHE.load()
    imgpath = input_face(args.camera)

    face_detection = FaceDetection()
    face_data = face_detection.handle(imgpath, True)

    face_alignment = FaceAlignment()
    face_alignment.handle(face_data)


if '__main__' == __name__ :
    test(args())