用pytorch实现MTCNN

我参考的是这个https://blog.csdn.net/qq_32478489/article/details/106193921

照他的步骤来，但是会有几个错误，我这里记录一下

1.按照他的运行detect.py

2.然后下载文件包，放在

3.然后先别往下训练Pnet

要自己在data_set中创建train，在train里面创建文件夹12，24，48

同时在anno_store里面自己创建part_12,等等以12为后缀的，他原文件夹好像没有

（选执行）如果电脑不想训练时间过长只是想跑通一下代码，建议直接把anno_train.txt中的只删除到一个大开头，比如我这里只留下了12

同时在下面图片路径下只留下12

 4.开始训练Pnet前

在他要运行的文件里面找到那些地址，改成自己路径，不要用原代码的 ./ ，同理在训练Rnet，Onet的时候也要这样，下图是其中一个文件的例子

这个问题在后面训练网络时有时候不会报错，但是在后续运行中会报错，所以一定要确保所要运行中的程序中，把路径都改好。

5.训练Pnet

会生成下图这些

 6.开始训练Rnet前

打开anno_train.txt，一定要把最后几个空白行删掉，不然会报错，不要留空白行

7.训练Rnet

注意4的问题

生成rnet_epoch_*.pt

8.训练Onet

注意4的问题

生成onet_epoch_*.pt

9.测试

最后在运行测试代码wildface_test.py时把他的模型地址改成自己的模型地址

 

结果肯定是没人家原始的好，毕竟我就训练了很少的数据集