代码环境:
Windows10,python 3.6,opencv 3.x,tensorflow-gpu,face-net 的 mtcnn 模块
mtcnn 代码下载:
https://github.com/davidsandberg/facenet/tree/master/src/align
代码:
import tensorflow as tf
import face_detect.align.detect_face as detect_face
import cv2
minsize = 20 # 脸部最小的大小
threshold = [0.6, 0.7, 0.7] # 三个步骤的阈值
factor = 0.709 # 用于在图像中检测的人脸大小的缩放金字塔的因子
# 分配给 tensorflow 的 gpu 的显存大小: GPU 实际显存 * 1.0
gpu_memory_fraction = 1.0
# 创建 tensorflow 网络并加载参数
with tf.Graph().as_default():
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=gpu_memory_fraction)
sess = tf.Session(config=tf.ConfigProto(gpu_optio