AI读脸术二次开发：模型替换与自定义标签实现教程

1. 引言

1.1 学习目标

本文旨在指导开发者如何对“AI读脸术”项目进行深度二次开发，重点涵盖模型替换机制与自定义标签系统实现。完成本教程后，您将能够：

理解 OpenCV DNN 模型加载与推理流程
替换默认的年龄、性别分类模型为自定义训练模型
修改或扩展输出标签内容（如增加情绪识别、佩戴眼镜等属性）
实现模型热更新与配置化管理

本教程适用于具备 Python 基础和一定计算机视觉经验的开发者。

1.2 前置知识

在开始前，请确保已掌握以下基础知识：

Python 编程语言基础语法
OpenCV 库的基本使用（图像读取、绘制）
深度学习中 Caffe 模型结构理解（.prototxt与.caffemodel）
HTTP 接口调用与 WebUI 基本交互逻辑

💡 温馨提示：本文所涉及的所有操作均在官方镜像基础上进行，无需额外安装依赖库。

2. 项目架构解析

2.1 整体架构概览

“AI读脸术”采用轻量级前后端一体化设计，核心组件包括：

前端界面：基于 Flask 提供的简易 WebUI，支持图片上传与结果展示
后端服务：Python + OpenCV DNN 实现人脸检测与属性分析
模型文件：三个独立的 Caffe 模型分别用于：
- res10_300x300_ssd_iter_140000.caffemodel：人脸检测
- age_net.caffemodel：年龄预测
- gender_net.caffemodel：性别分类

所有模型均已持久化至/root/models/目录，避免重启丢失。

2.2 核心模块职责划分

模块	路径	功能说明
主入口	`/app/app.py`	Flask 启动服务，处理上传请求
图像处理	`/app/utils.py`	图像预处理、DNN 推理封装
模型配置	`/app/config.py`	模型路径、标签映射表定义
静态资源	`/app/static/`	前端页面与样式文件

3. 模型替换实践指南

3.1 准备自定义模型

若需替换原有模型，必须满足以下条件：

模型格式为 Caffe 架构（.prototxt+.caffemodel）
输入尺寸与原模型一致（通常为 227×227 或 224×224）
输出层维度匹配原标签数量（性别：2类；年龄：8类）

示例：使用自定义性别模型

假设你已训练好一个更精准的性别分类模型，命名为：

/custom_models/gender.prototxt
/custom_models/gender.caffemodel

将其上传至服务器并移动到模型目录：

mv /custom_models/*.caffemodel /root/models/ mv /custom_models/*.prototxt /root/models/

3.2 修改模型加载逻辑

打开配置文件/app/config.py，找到模型路径定义部分：

# config.py MODEL_PATHS = { "face": "/root/models/res10_300x300_ssd_iter_140000.caffemodel", "face_proto": "/root/models/deploy.prototxt", "age": "/root/models/dex_chalearn_iccv2015.caffemodel", "age_proto": "/root/models/age_deploy.prototxt", "gender": "/root/models/gender.caffemodel", "gender_proto": "/root/models/gender_deploy.prototxt" }

将"gender"和"gender_proto"修改为新模型路径：

"gender": "/root/models/gender.caffemodel", "gender_proto": "/root/models/gender.prototxt"

3.3 验证模型兼容性

启动服务前，建议先测试模型是否可正常加载：

import cv2 net = cv2.dnn.readNetFromCaffe("/root/models/gender.prototxt", "/root/models/gender.caffemodel") print("✅ 自定义性别模型加载成功")

如无报错，则表示模型结构正确，可以继续部署。

4. 自定义标签系统实现

4.1 理解标签映射机制

当前项目的性别与年龄标签由硬编码列表控制。查看/app/utils.py中的相关代码：

# utils.py 片段 GENDER_LIST = ['Male', 'Female'] AGE_INTERVALS = ['(0-2)', '(4-6)', '(8-12)', '(15-20)', '(25-32)', '(38-43)', '(48-53)', '(60-100)']

这些列表决定了模型输出索引对应的语义标签。

4.2 扩展标签类别（以添加情绪识别为例）

步骤一：准备情绪识别模型

假设你已拥有一个基于 FER2013 训练的情绪分类模型：

/root/models/emotion.prototxt
/root/models/emotion.caffemodel
输出标签：['Angry', 'Disgust', 'Fear', 'Happy', 'Sad', 'Surprise', 'Neutral']

步骤二：修改配置文件

在config.py中新增情绪模型路径：

"emotion": "/root/models/emotion.caffemodel", "emotion_proto": "/root/models/emotion.prototxt"

步骤三：更新工具函数

在utils.py中添加情绪推理函数：

def predict_emotion(face_roi, net): blob = cv2.dnn.blobFromImage(face_roi, 1.0, (64, 64), (104, 177, 123)) net.setInput(blob) pred = net.forward() label_id = pred[0].argmax() confidence = pred[0][label_id] return EMOTION_LABELS[label_id], confidence EMOTION_LABELS = ['Angry', 'Disgust', 'Fear', 'Happy', 'Sad', 'Surprise', 'Neutral']

步骤四：集成至主流程

在app.py的主处理函数中加入情绪识别逻辑：

# 加载情绪模型 emotion_net = cv2.dnn.readNetFromCaffe(config.MODEL_PATHS["emotion_proto"], config.MODEL_PATHS["emotion"]) # 在人脸循环中添加 for (x, y, w, h) in faces: face_crop = image[y:y+h, x:x+w] gender, gender_conf = predict_gender(face_crop, gender_net) age, age_conf = predict_age(face_crop, age_net) emotion, emo_conf = predict_emotion(cv2.resize(face_crop, (64,64)), emotion_net) label = f"{gender}, {age}, {emotion}" cv2.putText(image, label, (x, y-10), cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0,255,0), 2)

重启服务后，即可在图像上看到包含情绪信息的新标签。

5. 工程优化与最佳实践

5.1 模型热更新机制

为避免每次更换模型都需要重启服务，可实现简单的模型重载接口：

@app.route('/reload_model', methods=['POST']) def reload_model(): global gender_net, age_net try: gender_net = cv2.dnn.readNetFromCaffe(config.MODEL_PATHS["gender_proto"], config.MODEL_PATHS["gender"]) age_net = cv2.dnn.readNetFromCaffe(config.MODEL_PATHS["age_proto"], config.MODEL_PATHS["age"]) return {"status": "success", "message": "模型重载完成"} except Exception as e: return {"status": "error", "message": str(e)}, 500

通过 POST 请求即可动态切换模型。