项目简介
本项目是基于Python和百度AI构建的人脸识别智能相册。借助调用百度AI的人脸识别API以及自研的U2GE - Net显著性检测网络,可实现对指定人物照片的智能筛选。能识别图像里的人脸,通过显著性检测确定图像主体,最终依据人脸与主体的重合度来决定照片是否保存。
项目的主要特性和功能
- 人脸识别:调用百度AI的人脸识别API,识别图像中的人脸并返回人脸预测框。
- 显著性检测:运用U2GE - Net网络对图像进行显著性检测,生成预测掩码以确定图像主体区域。
- 智能筛选:通过计算人脸预测框与显著性检测掩码的重合度,筛选出以指定人物为主体的照片。
- 参数可调:支持调整最大识别人脸数量和置信度阈值,适配不同应用场景。
安装使用步骤
1. 安装依赖
首先,确保已安装Python环境,并安装百度AI的Python SDK。
bash
pip install baidu-aip
2. 配置API密钥
在face_recognition.py
文件中配置百度AI的API密钥。
3. 运行项目
运行face_recognition.py
文件,开始处理图像。
bash
python face_recognition.py
4. 调整参数
根据需要,可以在face_recognition.py
文件中调整以下参数:
- options['max_face_num']
:设置最大识别人脸的数量(不超过10个)。
- options['match_threshold']
:设置人脸识别的置信度阈值。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】