Skip to content

huangshenneng/padd_retinaface

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published