Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
hypo
2eb5f6ac4e
|
2 years ago | |
|---|---|---|
| script | 2 years ago | |
| src | 2 years ago | |
| README.md | 2 years ago | |
| eval_ijbc.py | 2 years ago | |
| export.py | 2 years ago | |
| requirements.txt | 2 years ago | |
| train.py | 2 years ago | |
| val.py | 2 years ago | |
README.md
目录
Arcface概述
使用深度卷积神经网络进行大规模人脸识别的特征学习中的主要挑战之一是设计适当的损失函数以增强判别能力。继SoftmaxLoss、Center Loss、A-Softmax Loss、Cosine Margin Loss之后，Arcface在人脸识别中具有更加良好的表现。Arcface是传统softmax的改进， 将类之间的距离映射到超球面的间距，论文给出了对此的清晰几何解释。同时，基于10多个人脸识别基准的实验评估，证明了Arcface优于现有的技术，并且可以轻松实现。
论文： Deng J , Guo J , Zafeiriou S . ArcFace: Additive Angular Margin Loss for Deep Face Recognition[J]. 2018.
数据集
使用的训练数据集：MS1MV2
验证数据集：lfw，cfp-fp，agedb，cplfw，calfw，IJB-B，IJB-C
训练集：5,822,653张图片，85742个类
环境要求
-
硬件：昇腾处理器（Ascend）
- 使用Ascend处理器来搭建硬件环境。
-
框架
-
如需查看详情，请参见如下资源：
快速入门
通过官方网站安装MindSpore后，您可以按照如下步骤进行训练和评估：
# 分布式训练运行示例
sh scripts/run_distribute_train.sh rank_size /path/dataset
# 单机训练运行示例
sh scripts/run_standalone_train.sh /path/dataset
# 运行评估示例
sh scripts/run_eval.sh /path/evalset /path/ckpt
脚本说明
脚本和样例代码
└── Arcface
├── README.md // Arcface相关描述
├── scripts
├── run_distribute_train.sh // 用于分布式训练的shell脚本
├── run_standalone_train.sh // 用于单机训练的shell脚本
├── run_eval_ijbc.sh // 用于IJBC数据集评估的shell脚本
└── run_eval.sh // 用于评估的shell脚本
├──src
├── export.py
├── loss.py //损失函数
├── dataset.py // 创建数据集
├── iresnet.py // ResNet架构
├──val.py // 测试脚本
├──train.py // 训练脚本
脚本参数
train.py和val.py中主要参数如下：
-- modelarts：是否使用modelarts平台训练。可选值为True、False。默认为False。
-- device_id：用于训练或评估数据集的设备ID。当使用train.sh进行分布式训练时，忽略此参数。
-- device_num：使用tra进行分布式训练时使用的设备数。
-- train_url：checkpoint的输出路径。
-- data_url：训练集路径。
-- ckpt_url：checkpoint路径。
-- eval_url：验证集路径。
训练过程
分布式训练
sh scripts/run_distribute_train.sh rank_size /path/dataset
上述shell脚本将在后台运行分布训练。可以通过device[X]/train.log文件查看结果。
采用以下方式达到损失值：
epoch: 2 step: 11372, loss is 12.807039
epoch time: 1104549.619 ms, per step time: 97.129 ms
epoch: 3 step: 11372, loss is 9.13787
...
epoch: 21 step: 11372, loss is 1.5028578
epoch time: 1104673.362 ms, per step time: 97.140 ms
epoch: 22 step: 11372, loss is 0.8846929
epoch time: 1104929.793 ms, per step time: 97.162 ms
评估过程
评估
-
在Ascend环境运行时评估lfw、cfp_fp、agedb_30、calfw、cplfw数据集
在运行以下命令之前，请检查用于评估的检查点路径。请将检查点路径设置为绝对全路径，例如“username/arcface/arcface-11372-1.ckpt”。
sh scripts/run_eval.sh /path/evalset /path/ckpt上述python命令将在后台运行，您可以通过eval.log文件查看结果。测试数据集的准确性如下：
[lfw]Accuracy-Flip: 0.99817+-0.00273 [cfp_fp]Accuracy-Flip: 0.98000+-0.00586 [agedb_30]Accuracy-Flip: 0.98100+-0.00642 [calfw]Accuracy-Flip: 0.96150+-0.01099 [cplfw]Accuracy-Flip: 0.92583+-0.01367 -
在Ascend环境运行时评估IJB-B、IJB-C数据集
在运行以下命令之前，请检查用于评估的检查点路径。请将检查点路径设置为绝对全路径，例如“username/arcface/arcface-11372-1.ckpt”。
同时，情确保传入的评估数据集路径为“IJB_release/IJBB/”或“IJB_release/IJBC/”。
sh scripts/run_eval_ijbc.sh /path/evalset /path/ckpt上述python命令将在后台运行，您可以通过eval.log文件查看结果。测试数据集的准确性如下：
+-----------+-------+-------+--------+-------+-------+-------+ | Methods | 1e-06 | 1e-05 | 0.0001 | 0.001 | 0.01 | 0.1 | +-----------+-------+-------+--------+-------+-------+-------+ | ijbb-IJBB | 40.01 | 87.91 | 94.36 | 96.48 | 97.72 | 98.70 | +-----------+-------+-------+--------+-------+-------+-------+ +-----------+-------+-------+--------+-------+-------+-------+ | Methods | 1e-06 | 1e-05 | 0.0001 | 0.001 | 0.01 | 0.1 | +-----------+-------+-------+--------+-------+-------+-------+ | ijbc-IJBC | 82.08 | 93.37 | 95.87 | 97.40 | 98.40 | 99.05 | +-----------+-------+-------+--------+-------+-------+-------+
模型描述
性能
评估性能
| 参数 | Arcface |
|---|---|
| 模型版本 | arcface |
| 资源 | Ascend 910； CPU： 2.60GHz，192内核；内存，755G |
| 上传日期 | 2021-05-30 |
| MindSpore版本 | 1.2.0-c77-python3.7-aarch64 |
| 数据集 | MS1MV2 |
| 训练参数 | lr=0.08; gamma=0.1 |
| 优化器 | SGD |
| 损失函数 | Arcface |
| 输出 | 概率 |
| 损失 | 0.6 |
| 速度 | 1卡：108毫秒/步；8卡：97毫秒/步 |
| 总时间 | 1卡：65小时；8卡：8.5小时 |
| 参数(M) | 85.2 |
| 微调检查点 | 1249M （.ckpt file） |
| 脚本 | 脚本路径 |
推理性能
| 参数 | Arcface |
|---|---|
| 模型版本 | arcface |
| 资源 | Ascend 910 |
| 上传日期 | 2021/05/30 |
| MindSpore版本 | 1.2.0-c77-python3.7-aarch64 |
| 数据集 | IJBC、IJBB、lfw、cfp_fp、agedb_30、calfw、cplfw |
| 输出 | 概率 |
| 准确性 | lfw:0.998 cfp_fp:0.98 agedb_30:0.981 calfw:0.961 cplfw:0.926 IJB-B:0.943 IJB-C:0.958 |
使用方法
推理
如果您需要使用已训练模型在GPU、Ascend 910、Ascend 310等多个硬件平台上进行推理，可参考此处。
迁移学习
待补充
随机情况说明
网络的初始参数均为随即初始化。
ModelZoo主页
请浏览官网主页。
No Description
Python Shell Text