Skip to content

GitLab

Commit 559bf197 authored by dongchy920's avatar dongchy920
Browse files

Update README.md

parent 1abbb577
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
README.md
View file @ 559bf197
...@@ -88,6 +88,9 @@ unzip IJBC_ALL.zip ...@@ -88,6 +88,9 @@ unzip IJBC_ALL.zip
SCNet下载链接[IJBC.zip](http://113.200.138.88:18080/aidatasets/project-dependency/arcface_pytorch)，通过SCNet下载的压缩文件解压并将文件夹重命名为IJBC_ALL 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)，SCNet下载链接[model.pt](http://113.200.138.88:18080/aidatasets/project-dependency/arcface_pytorch) 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)
在对应configs文件中修改模型保存路径和训练数据的加载路径，比如说：
在configs/ms1mv2_r100.py中修改数据加载路径[config.rec](./configs/ms1mv2_r100.py#L22)
在configs/base.py中修改模型保存路径[config.output](./configs/base.py#L14)，模型名称可以在train_v2.py中修改，默认为[model.pt](train_v2.py#L228)
### 单机单卡 ### 单机单卡
``` ```
......
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