【源码】基于PyTorch框架的人脸识别模型训练与评估系统

项目简介

本项目基于PyTorch框架，专注于人脸识别模型的训练与评估。覆盖了从数据预处理到模型训练、评估的全流程，方便用户搭建人脸识别模型并开展相关工作。

项目的主要特性和功能

1. 支持数据预处理，包含图像加载、尺寸调整、色彩空间转换及归一化等操作。
2. 定义基于ResNet的人脸识别模型Facenet_V2，有多种预定义模型配置。
3. 提供模型训练脚本，可使用GPU或CPU计算，支持多种优化器，能保存训练好的模型权重。
4. 具备模型评估功能，可计算ROC AUC分数等评估指标。

安装使用步骤

安装依赖

使用pip安装项目所需的PyTorch、OpenCV等库。

数据准备

准备好包含图像路径和标签信息的CSV文件。

模型训练

运行以下命令进行模型训练： bash python train.py ../init_data/toUser/train model-best.pth

模型测试

运行以下命令进行模型测试： bash python test.py ../init_data/toUser/test/data result.csv

下载地址

点击下载 【提取码: 4003】【解压密码: www.makuang.net】