准备Rnet需要的box - SiYue0211/mtcnn_tf GitHub Wiki
准备RNet需要的Box
-
找到PNet训练model里的最后一个checkpoint,将其训练好的参数载入Pnet模型里。
-
读取Wider数据集,将每张图片地址和图片对应的box放入一个词典
-
用PNet的参数测试一边Pnet所有图片,生成候选框。并将其pickle到一个指定文件
-
将3生成的候选框(每个图片有一堆)与gt box计算IOU,大于0.65为正例子,在0.4到0.65之间的为可忽视例子,小于0.3为负例子