人脸识别毕设从零到一：新手入门技术选型与避坑指南-程序员充电站

人脸识别毕设从零到一：新手入门技术选型与避坑指南

摘要：很多本科同学第一次做人脸识别毕设，都会卡在“模型选哪个”“环境怎么搭”“准确率上不去”这三座大山。本文用通俗语言把常见坑位一次性标好，再给出一条“Python + OpenCV + Flask”的轻量路线，附带可直接跑的代码模板，帮你两周内拿出能演示的毕业设计。

一、先吐槽：为什么人脸识别毕设总翻车？

模型泛化差
实验室拍的头像光照均匀、角度端正，一换到教室窗边就全挂。
依赖复杂
今天cuda==11.1，明天torch==1.9，后天系统升级，环境直接炸。
无法部署
笔记本上 99% 准确率，拷到老师离线电脑就提示缺 DLL，演示现场社死。

一句话：选题时想的是“高大上”，落地时全是“环境坑”。下面按“选模型 → 写代码 → 跑服务 → 避坑”四步，带你把毕设拉回可控范围。

二、技术选型：四款开源方案 5 维对比

方案	精度(1:1)	速度(CPU)	依赖重量	学习曲线	适合场景
OpenCV内置LBP	0.825	15 ms	零依赖	★	课堂演示、极低配
Dlib CNN	0.921	120 ms	仅dlib	★★	本地小项目
FaceNet	0.965	80 ms	torch+mtcnn	★★★	轻量论文复现
InsightFace	0.993	40 ms	mxnet/onnx	★★★★	追求SOTA、有GPU

结论：本科毕设建议FaceNet——精度够用、代码好读、CPU 也能跑；如果实验室有 3060 以上显卡，再考虑 InsightFace。

三、核心实现：30 行代码跑通“检测+对齐+特征+比对”

下面代码全部单文件，按 Clean Code 拆函数，复制即可跑。

环境一键安装

# 创建虚拟环境 python -m venv face_grad face_grad\Scripts\activate pip install opencv-python dlib tensorflow==2.10 facenet-pytorch flask

项目骨架

face_grad/ ├─ app.py # Flask 接口 ├─ face_engine.py # 核心逻辑 ├─ static/ # 上传照片 └─ model/ └─ facenet512.h5 # 预训练权重（自己下载）

face_engine.py（关键步骤已写注释）

import cv2, dlib, numpy as np from tensorflow.keras.models import load_model DETECTOR = dlib.get_frontal_face_detector() SP = dlib.shape_predictor("model/shape_68.dat") MODEL = load_model("model/facenet512.h5") # 512维特征 def align_face(img): """返回 160×160 对齐后人脸""" gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY) rects = DETECTOR(gray, 1) if len(rects) == 0: return None shape = SP(gray, rects[0]) # 计算仿射变换矩阵，这里简化：双眼水平 le, re = shape.part(36), shape.part(45) dy, dx = re.y - le.y, re.x - le.x angle = np.degrees(np.arctan2(dy, dx)) center = ((le.x + re.x) // 2, (le.y + re.y) // 2) M = cv2.getRotationMatrix2D(center, angle, 1) warped = cv2.warpAffine(img, M, (img.shape[1], img.shape[0])) # 裁剪 x, y, w, h = rects[0].left(), rects[0].top(), rects[0].width(), rects[0].height() face = warped[y:y + h, x:x + w] return cv2.resize(face, (160, 160)) def encode(img): """返回 512 维特征向量""" face = align_face(img) if face is None: return None face = (face / 255.).astype('float32') vec = MODEL.predict(np.expand_dims(face, axis=0))[0] return vec / np.linalg.norm(vec) # L2 归一化 def cosine_similarity(v1, v2): return np.dot(v1, v2)

Flask 简易 API（app.py）

from flask import Flask, request, jsonify from face_engine import encode, cosine_similarity import cv2, numpy as np app = Flask(__name__) db = {} # 内存级“人脸库”，key=姓名，value=特征向量 @app.route('/register', methods=['POST']) def register(): file = request.files['image'] name = request.form['name'] img = cv2.imdecode(np.frombuffer(file.read(), np.uint8), 1) vec = encode(img) if vec is None: return "no face", 400 db[name] = vec return "ok", 200 @app.route('/verify', methods=['POST']) def verify(): file = request.files['image'] img = cv2.imdecode(np.frombuffer(file.read(), np.uint8), 1) vec = encode(img) if vec is None: return "no face", 400 scores = {n: cosine_similarity(vec, v) for n, v in db.items()} best, score = max(scores.items(), key=lambda x: x[1]) return jsonify({"name": best if score > 0.45 else "unknown", "score": float(score)})

启动服务：

python app.py # 打开 http://127.0.0.1:5000 即可上传注册/验证

四、性能与安全：本地跑起来后，还要想三件事

资源消耗
- FaceNet 512 模型 93 M，冷启动 2.3 s，常驻内存 230 M；CPU 单张 80 ms，足够实时。
延迟优化
- 把dlib.shape_predictor提前加载到内存，避免每次重建。
- Flask 开threaded=True，并发 4 路以内无压力。
隐私保护
- 上传图片先缩放 ≤ 1080p，再随机文件名保存，演示结束立即os.remove。
- 特征向量不可逆，可放心落库；但原始照片务必加密或定期清理，防止泄露。

五、生产环境避坑指南（血泪版）

模型路径硬编码
换电脑就报错？用pathlib.Path(__file__).parent / "model"动态获取。
多张人脸未处理
注册只取rects[0]，验证同理；若出现多人脸，返回提醒“请保持单人入镜”。
光照敏感
对齐前加cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8, 8))做直方图均衡，户外准确率能提 5-8%。
跨姿态翻车
侧脸 > 30° 时 68 点 landmark 会飘，考虑用 InsightFace 的 2D106 点模型，或干脆拒绝大角度。
版本地狱
把requirements.txt钉小版本，并在 README 写明“Python 3.8+ 64 位、VS BuildTools 2019 以上”，给老师省麻烦。