Mask-Face-Recognition:口罩人脸识别
●该项目主要是实现人脸特征向量的提取。以标准人脸识别模型FaceNet为主线,添加fpn_face_attention结构,增加CBAM模块,可以更好地聚焦于人面部半部,没带口罩的区域
整理数据集
正常人脸训练数据:VGGFace2,链接: ://www.robots.ox.ac.uk/~vgg/data/vgg_face2/正常人脸测试数据:LFW(狂野标签脸),链接: : pair.txt: ://vis-www.cs.umass.edu/lfw/pairs.txt(pairs.txt文件是图片对文件,包含测试的图片对,以及标注)
首先生成一个数据集文件夹,把VGGFace2的原始数据(VGGFace2_train文件),LFW原始数据(lfw_funneled),LFW耦合文件(LFW_pairs.txt),都放到数据
2022-08-08 21:29:57
894KB
Python
1