Commit 1abbb577 authored by dongchy920's avatar dongchy920
Browse files

Update icon.png, requirements.txt, model.properties, README.md files

parent 06465b3d
Pipeline #1852 failed with stages
in 0 seconds
...@@ -23,7 +23,7 @@ ArcFace训练流程: ...@@ -23,7 +23,7 @@ ArcFace训练流程:
### Docker(方法一) ### Docker(方法一)
[光源](https://www.sourcefind.cn/#/service-list)中拉取docker镜像: [光源](https://www.sourcefind.cn/#/service-list)中拉取docker镜像:
``` ```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk23.10-py310 docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.2-py3.10
``` ```
创建容器并挂载目录进行开发: 创建容器并挂载目录进行开发:
``` ```
...@@ -50,10 +50,10 @@ conda activate arcface ...@@ -50,10 +50,10 @@ conda activate arcface
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
``` ```
DTK驱动:dtk23.10 DTK驱动:dtk24.04.2
python:python3.10 python:python3.10
pytorch:1.13.1 pytorch:2.1.0
torchvision:0.14.1 torchvision:0.16.0
``` ```
安装其他依赖包 安装其他依赖包
``` ```
...@@ -62,7 +62,7 @@ pip install -r requirements.txt ...@@ -62,7 +62,7 @@ pip install -r requirements.txt
## 数据集 ## 数据集
`MS1MV2\IJBC` `MS1MV2\IJBC`
- 训练集[faces_emore.zip](https://pan.baidu.com/s/1S6LJZGdqcZRle1vlcMzHOQ) - 训练集[faces_emore.zip](https://pan.baidu.com/s/1S6LJZGdqcZRle1vlcMzHOQ)SCNet下载链接[faces_emore.zip](http://113.200.138.88:18080/aidatasets/project-dependency/arcface_pytorch)
下载后解压到当前目录 下载后解压到当前目录
数据目录结构如下: 数据目录结构如下:
``` ```
...@@ -78,16 +78,16 @@ pip install -r requirements.txt ...@@ -78,16 +78,16 @@ pip install -r requirements.txt
| train.rec | train.rec
| vgg2_fp.bin | vgg2_fp.bin
``` ```
- 测试集[IJBC.zip](https://pan.baidu.com/s/1Ok4sqTO8vqAE_kG3zV1rqw?pwd=1234) - 测试集[IJBC.zip](https://pan.baidu.com/s/1Ok4sqTO8vqAE_kG3zV1rqw?pwd=1234),
解压分卷压缩文件: 解压分卷压缩文件:
``` ```
# 将所有的分卷压缩文件放在一个文件夹中 # 将所有的分卷压缩文件放在一个文件夹中
zip -s 0 IJBC.zip --out IJBC_ALL.zip zip -s 0 IJBC.zip --out IJBC_ALL.zip
unzip IJBC_ALL.zip unzip IJBC_ALL.zip
``` ```
SCNet下载链接[IJBC.zip](http://113.200.138.88:18080/aidatasets/project-dependency/arcface_pytorch),通过SCNet下载的压缩文件解压并将文件夹重命名为IJBC_ALL
## 训练 ## 训练
Backbone使用ResNet100,在MS1MV3数据集上的预训练权重文件为[model.pt](https://pan.baidu.com/s/1W-TisIZtZmRQz32hq5T6Uw?pwd=1234) Backbone使用ResNet100,在MS1MV3数据集上的预训练权重文件为[model.pt](https://pan.baidu.com/s/1W-TisIZtZmRQz32hq5T6Uw?pwd=1234),SCNet下载链接[model.pt](http://113.200.138.88:18080/aidatasets/project-dependency/arcface_pytorch)
### 单机单卡 ### 单机单卡
``` ```
......
icon.png

75.2 KB

...@@ -3,7 +3,7 @@ modelCode=676 ...@@ -3,7 +3,7 @@ modelCode=676
# 模型名称 # 模型名称
modelName=arcface_pytorch modelName=arcface_pytorch
# 模型描述 # 模型描述
modelDescription=ArcFace设计了新的用于人脸识别的损失函数, modelDescription=ArcFace设计了新的用于人脸识别的损失函数, 通过引入角度间隔,使模型学习到具有更好区分能力的特征表示,从而提升识别精度
# 应用场景 # 应用场景
appScenario=推理,训练,人脸识别,安防,交通,教育 appScenario=推理,训练,人脸识别,安防,交通,教育
# 框架类型 # 框架类型
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment