【备忘】insightface使用自定义数据集

insightface

每个人id一个文件夹的图片

• 每个人一个文件夹的图片数据集（raw-dataset）；

• 对raw dataset 进行脸部检测与对其裁剪为112x112的图片，得到face-dataset；

  • 使用alignCropFace.py函数来检测脸和对齐裁剪

  注意其中对于每张图检测出来单张脸或多于一张脸的处理部分。

   python alignCropFace.py --src_dir /path/to/src/dir --crop_dir /path/to/tar_dir --nc 1 --gpuid 1 --thres 0.9
  

• 对raw dataset进行口罩佩戴，得到裁剪后对口罩脸（mask-dataset）;
  python3 wearmask.py --src_dir /path/to/src/dir --dst_dir /path/to/tar_dir --nc 10

• 按人名id合并 face-dataset和mask-dataset, 得到mixed-datset;

• 生成.lst文件

  • 使用dir2lst.py对裁剪好对脸图数据集生成列表文件;
  • python dir2lst.py > youlistfile.lst;
  • lst文件包含脸图片是否对齐，图片路径，标签对信息；

• 使用face2rec2.py生成.rec文件；

  • 在工作文件夹内先创建property文件: 手动创建，比如 93979,112,112；
  • 保证工作文件夹内有*.lst文件夹；
  • python face2rec2.py --prefix /path/to/save/lst/file --encoding .jpg --recursive True