我参考的是这个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时把他的模型地址改成自己的模型地址
结果肯定是没人家原始的好,毕竟我就训练了很少的数据集