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,所以我们要自己转一下(