5.0 KiB
5.0 KiB
照片管理系统项目文档
项目概述
本项目是一个基于 Flask 框架的照片管理系统,主要功能包括:
- 人脸管理:增删改查用户的人脸数据。
- 照片上传与分类:用户可以上传照片,系统会根据 EXIF 时间信息和数据库中的人脸数据进行分类存储。
- 照片查询:用户可以根据姓名和日期查找相应的照片。
技术栈
- 后端:Flask(Python)
- 数据库:SQLAlchemy(SQLite / MySQL)
- 前端:HTML + CSS + Jinja2
- 存储:文件系统存储上传的照片
数据库设计
1. users
表(用户信息表)
字段名 | 类型 | 说明 |
---|---|---|
id | 主键 | 自动生成 |
username | 字符串 | 用户名,唯一且非空 |
encoding | 二进制数据 | 人脸识别编码,用于匹配 |
photo | 字符串 | 用户头像照片路径 |
2. photos
表(照片信息表)
字段名 | 类型 | 说明 |
---|---|---|
id | 主键 | 自动生成 |
user_id | 外键 | 关联到 users 表的 id |
created_at | 时间戳 | 照片拍摄时间(来自 EXIF 信息) |
classification | 布尔值 | 照片是否分类(默认 False ,分类成功后 True ) |
classification_confidence | 浮点数 | 照片分类置信度(默认 0 ) |
classification_image_path | 字符串 | 分类后的照片存储路径 |
主要功能实现
1. 用户管理
- 添加用户:用户可上传照片,系统提取人脸编码并存入数据库。
- 查看用户列表:显示所有用户及其头像。
- 删除用户:从数据库中删除用户信息及相关照片。
2. 照片上传与分类
- 上传照片:用户上传照片,系统解析 EXIF 数据确定拍摄时间。
- 人脸匹配分类:系统根据数据库中的人脸编码进行匹配,分类并存储到相应目录。
- 存储路径规则:
static/uploads/users/{用户名}/{日期(年-月-日)}/照片文件名
3. 照片查询
- 基于姓名和日期查询:用户输入姓名和日期,系统返回符合条件的照片。
- 展示分类结果:前端页面显示照片缩略图,并提供原始照片下载链接。
关键代码
1. 照片上传(upload
端点)
@app.route('/upload', methods=['GET', 'POST'])
def upload():
if request.method == 'POST':
photos = request.files.getlist('photos')
classify_photos(photos) # 进行分类处理
# 获取最新照片并关联用户信息
recent_photos = Photo.query.order_by(Photo.created_at.desc()).limit(len(photos)).all()
photos_with_users = []
for photo in recent_photos:
user = User.query.get(photo.user_id)
photos_with_users.append({
"filename": photo.classification_image_path,
"username": user.username if user else "未知",
"created_at": photo.created_at
})
return render_template('upload.html', photos=photos_with_users)
return render_template('upload.html', photos=[])
2. 照片查询(search
端点)
@app.route('/search', methods=['POST'])
def search():
name = request.form['name']
date = request.form['date']
user = User.query.filter_by(username=name).first()
if not user:
return render_template('search.html', photos=[], message="用户不存在")
photos = Photo.query.filter_by(user_id=user.id).filter(Photo.created_at.like(f"{date}%")).all()
return render_template('search.html', photos=photos)
前端页面示例
1. 照片查询页面(search.html
)
<form action="{{ url_for('search') }}" method="POST">
<label for="name">姓名:</label>
<input type="text" id="name" name="name" placeholder="请输入用户名" required>
<label for="date">日期:</label>
<input type="date" id="date" name="date" required>
<button type="submit">查询照片</button>
</form>
目录结构
project_root/
│── app.py # Flask 入口文件
│── functions.py # 处理照片分类等功能
│── models.py # 数据库模型
│── templates/
│ ├── base.html # 公共模板
│ ├── upload.html # 上传页面
│ ├── search.html # 查询页面
│── static/
│ ├── uploads/ # 存储照片
│ ├── styles.css # 样式表
│── .venv/ # 虚拟环境
│── requirements.txt # 依赖文件
部署方式
- 安装依赖
pip install -r requirements.txt
- 运行 Flask 服务器
flask run
- 访问应用
在浏览器中打开
http://127.0.0.1:5000
未来优化方向
- 添加身份验证:使用 Flask-Login 实现用户登录功能。
- 优化分类算法:提高人脸识别和分类的准确率。
- 支持云存储:将照片存储到云端,如 AWS S3 或阿里云 OSS。
本项目旨在提供一个高效、易用的照片管理系统,欢迎优化和扩展!