insightface测试megaface流程

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/jmu201521121021/article/details/81151758

insightface github地址: https://github.com/deepinsight/insightface
megaface database地址: http://megaface.cs.washington.edu/

一、简介

  • insightface 测试megaface数据集的流程:
    • 对megaface和 facescrub数据集中的人脸对齐
    • 提取megaface和facescrub中对齐后人脸的特征
    • 移除megaface和facescrub中不干净的数据特征
    • 进行megaface测试

二、环境准备

  • mxnet-cu80
  • tensorflow
  • openCV2.4

三、具体步骤

  • 对megaface和 facescrub数据集中的人脸对齐
cd $INSIGHTFACE_ROOT/src/align/
# 由于facescrub中的部分图片没有后缀,所以要修改图片的名称。
python align_facescrub.py --input-dir='你的facescrubs数据地址' --output-dir='对齐后保存结果地址' 
# align megaface
python align_megaface.py  --input-dir=你的megaface数据地址' --name=' megaface' --output-dir='对齐后保存结果地址' 
  • 提取megaface和 facescrub对齐后的人脸特征
cd $INSIGHTFACE_ROOT/src/megaface
# gen_mefaface.py中需要修改变量:
# parser.add_argument('--model', type=str, help='', default='../model/spherefacei-s60-p0_15_96_112_0,95')  ../model/spherefacei-s60-p0_15_96_112_0代表模型的地址和开头, 95代表第95个保存的模型
# megaface_out = '/raid5data/dplearn/megaface/MegaFace_Features' #输出特征的指定地址
# facescrub_out = '/raid5data/dplearn/megaface/FaceScrub_Features'
# megaface_lst = "/raid5data/dplearn/megaface/megaface_mtcnn_112x112/lst" 对齐后的文件地址下list
# facescrub_lst = "/raid5data/dplearn/megaface/facescrubr/small_lst"

python -u gen_megaface.py
  • 移除megaface和facescrub中不干净的数据特征
cd $INSIGHTFACE_ROOT/src/megaface
# remove_noises.py 中需要修改变量
# parser.add_argument('--suffix', type=str, help='', default='r100_cm_112x112') #算法名称 
# parser.add_argument('--megaface-lst', type=str, help='', default='/raid5data/dplearn/megaface/megaface_mtcnn_112x112/lst')
# parser.add_argument('--facescrub-lst', type=str, help='', default='/raid5data/dplearn/megaface/facescrubr/small_lst')
# parser.add_argument('--megaface-feature-dir', type=str, help='', default='/raid5data/dplearn/megaface/MegaFace_Features')
# parser.add_argument('--facescrub-feature-dir', type=str, help='', default='/raid5data/dplearn/megaface/FaceScrub_Features') )
# parser.add_argument('--megaface-feature-dir-out', type=str, help='', default='/opt/jiaguo/MegaFace_Features_cm')
# parser.add_argument('--facescrub-feature-dir-out', type=str, help='', default='/opt/jiaguo/FaceScrub_Features_cm')
python -u remove_noises.py
  • 测试megaface (注意openCV版本为2.4 !!!!
cd $devkit_ROOT/experiments
python run_experiment_uncrop.py -p $devkit_ROOT/templatelists/facescrub_uncropped_features_list.json $MegaFace_Features_cm_82的文件夹地址 $FaceScrub_Features_cm_82文件夹地址 _mxasmr50_112x112.bin  $保存结果的地址
# 其中 _mxasmr50_112x112.bin是算法的名称+ .bin,由remove_noises.py中的--suffix 参数决定

  • 测试结果

这里写图片描述

  • 生成的文件的目录

这里写图片描述

猜你喜欢

转载自blog.csdn.net/jmu201521121021/article/details/81151758
今日推荐