browserliu / CCF-BDCI2019-Multi-person-Face-Recognition-Competition-Baseline

CCF BDCI2019 多人种人脸识别 Baseline Pubilic LB 0.64+ （PyTorch）

CCF-BDCI2019-Multi-person-Face-Recognition-Competition-Baseline

✨赛题链接：CCF BDCI2019 多人种人脸识别

这份代码主要是基于PyTorch框架实现，利用 MageFace 训练的预训练模型，来对赛题的测试集进行测试。

应用的模型是 InsightFace： Addittive Angular Margin Loss for Deep Face Recognition.paper

---

😋 UPDATE Oct. 8

增加整合了训练部分代码即复现步骤

---

🏃 准备工作

1️⃣ 下载代码

git clone https://github.com/LcenArthas/CCF-BDCI2019-Multi-person-Face-Recognition-Competition-Baseline.git

2️⃣ 配置环境

• Ubantu16.04

• Python 3.6

• PyTorch 1.0.0

---

✨ 训练部分

1️⃣ 准备数据

🔸 在根目录下创建文件夹 /train_data/，并将训练文件夹 /training/ 放入其中：

mkdir train_data

🔸 准备数据：

cd data
python pre_data.py

2️⃣ 开始训练

python train.py

---

✨ 测试提交部分

1️⃣ 配置测试集文件

把测试集的图片文件夹 /test/ 置于根目录，把提交例样 submission_template.csv 同样置于根目录

2️⃣ 下载预训练模型

• SE-LResNet101E-IR

下载好的模型置于根目录

---

👏 开始！

python test_ccf.py

10分钟左右，输出做后的提交结果 submission_new.csv

😄希望对小伙伴们有所启发！欢迎Star！