FaceNet中人脸检测部分代码总结

前言:课题需要使用facenet,但是不想用自带的mtcnn检测网络,想要替换成别的。写这篇文章记录一下源码分析过程。

1. 初步

人脸检测流程中,首先获取到照片之后,需要对人脸位置进行定位,之后裁剪对齐放到人脸识别模型中。所以在

faces = face_recognition.identify(frame)

就是人脸识别的入口,我们跟进去。

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

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

2.未完待续

posted @ 2021-03-19 16:06  swan1024  阅读(135)  评论(0)    收藏  举报