FaceNet中人脸检测部分代码总结
前言:课题需要使用facenet,但是不想用自带的mtcnn检测网络,想要替换成别的。写这篇文章记录一下源码分析过程。
1. 初步
人脸检测流程中,首先获取到照片之后,需要对人脸位置进行定位,之后裁剪对齐放到人脸识别模型中。所以在

faces = face_recognition.identify(frame)
就是人脸识别的入口,我们跟进去。

可以看到,faces = self.detect.find_faces(image)首先需要通过find_faces获取的人脸框,也就是人脸检测。我们只需要将这部分替换成自己的检测网络即可。

在detect_face方法中,需要传入原始图像,及pnet、onet、和rnet以及阈值等,这些都是mtcnn模型需要用到的参数,因此我们将该部分替换即可。

浙公网安备 33010602011771号