padd_retinaface

Paddle复现 RetinaFace: Single-stage Dense Face Localisation in the Wild https://arxiv.org/pdf/1905.00641v2.pdf ) 论文

##代码结构

/Paddle_Retinaface
|-- data
|   |-- widerface（训练数据）
|      |-- train
|          |-- images
|          |-- label.txt
|      |-- test
|          |-- images
|          |-- label.txt
|      |-- val
|          |-- images
|          |-- label.txt
|          |-- wider_val.txt
|   |-- config.py （超参数设置）
|   |-- data_augment.py （数据增强）
|   |-- wider_face.py （数据加载类）
|--functions
|       |-- prior_box.py （先验框设置）
|-- modules
|       |-- multibox_loss.py（损失函数）
|-- models
|       |-- net.py
|       |-- resnet.py
|       |-- retinaface.py（网络定义）
|-- utils
|       |-- nms
|           |-- py_cpu_nms.py
|       |-- box_utils.py
|-- widerface_evaluate          
|       |-- evaluation.py
|       |-- setup.py
|-- train.py（训练）
|-- test_widerface.py （测试）

准备工作

1. 下载训练集和测试集，放到  ' data/widerface'文件夹下
2. 到  'widerface_evaluate' 文件中，运行命令行 '  python setup.py build_ext --inplace'
3. 下载验证集的结果，解压到 ' data/widerface'文件夹下，用于测试mAP
4.下载模型

模型文件

baidu cloud Password: 0ewb

训练log

baidu cloud Password: ypry

模型的训练

直接运行train.py 文件

模型的测试

1.运行 test_widerface.py  
2. cd ./widerface_evaluate
    python evaluation.py

模型训练结果

Easy   Val AP: 0.9402364106525578
Medium Val AP: 0.9208495951285756
Hard   Val AP: 0.8005163465688859

参考 https://github.com/biubug6/Pytorch_Retinaface