【人脸检测】复现Pytorch_Retinaface

https://github.com/biubug6/Pytorch_Retinaface是人脸识别的代码源地址

在对上诉代码进行复现

1.环境

ubuntu16.04

python3.6

cuda10.0

pytorch1.1与torchview都是从官网下载的直接下载命令

opencv也是默认直接pip，读取图片的

Cython是在最后进行评估的

2.数据准备

没有用官方的数据0，用的数据1

链接：来自谷歌云或百度云密码：Ruck

3.模型

 google cloud and baidu cloud Password: fstq

4.训练模型

可以直接用官方指令

CUDA_VISIBLE_DEVICES=0 python train.py --network mobile0.25

这里是训练mobilenet0.25，另外一个resnet据说比较难训练我也没有尝试

注意：要将模型解压到model文件夹下面

5.测试

代码可以对数据进行测试

python test_widerface.py --trained_model ./weights/mobilenet0.25_epoch_245.pth --network mobile0.25 -s=True

这里注意如果直接用别人的模型进行测试要自己创建一个weight文件夹，将模型放在文件夹下面

检测结果jpg与txt分别在results与widerface_evaluate/widerface_txt目录下，-s=True是为了展示图，也可以直接在代码test_widerface.py中修改28行默认改为True。

检测结果jpg与txt分别在results与widerface_evaluate/widerface_txt目录下。

目前只跑到了这一步

标签：Retinaface,--,模型,py,Pytorch,widerface,复现,txt,True
来源： https://blog.csdn.net/weixin_44162984/article/details/118245458