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

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

Primary LanguagePythonMIT LicenseMIT

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️⃣ 下载预训练模型

下载好的模型置于根目录

👏 开始！

python test_ccf.py

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

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