【人脸检测】复现Pytorch_Retinaface(Pytorch版本)

https://github.com/biubug6/Pytorch_Retinaface

训练这个模型很耗GPU,mobilenet0.25还好(训练8G以内,测试1G以内),resnet50至少要43G。

0.环境

ubuntu16.04
python3.6
cuda9.0
torch==1.1.0
ipython

1.准备数据与模型

(1)准备数据

原始数据:train+val+test http://shuoyang1213.me/WIDERFACE/WiderFace_Results.html

标签数据:https://pan.baidu.com/s/1Laby0EctfuJGgGMgRRgykA

按照官方给的格式放:

./data/widerface/
    train/
      images/
      label.txt
    val/
      images/
      label.txt
    test/
      images/
      label.txt
    

由于val只有label.txt,所以我们要自己转一下(这一步一定要转一下,不然会测

你可能感兴趣的:(人脸算法,目标检测,RetinaFace,pytorch)