目录

(一) 人脸数据集对齐

(二) 生成lst文件


(一) 人脸数据集对齐

首先下载insightface源码,点击github链接下载,本文代码需在python2.7环境下运行,文章中所需的脚本文件点击下载

在进行训练之前。首先要准备好自己的数据,数据的存放形式如下图所示,每个文件夹代表一个人,文件夹下放置属于这个人的多张照片。

然后,需要对所有的照片中的人脸进行对齐处理,并把每张图片缩放到112x112。人脸对齐使用MTCNN,insightface中自带的就有MTCNN和其模型文件,拷贝face_alignment_by_mtcnn.py到deploy文件夹下,修改原始人脸数据文件夹和对齐后要保存的文件夹位置,如下图所示,然后直接运行该程序。

 

(二) 生成lst文件

lst文件中主要包含图片的路径、标签、序号等信息,为后面制作数据文件做准备,lst文件中的信息如下图所示,第一列时打乱的序号,第二列时图片的地址,第三列是该图片所对应的的标签。

开始制作。首先复制face2rec2_m.py文件到scr/data文件夹下,然后命令行窗口中输入以下命令:

python -u face2rec2_m.py --list True --prefix F:\datasets\20_faces_datasets\ --root F:\datasets\20_faces_clip\ --train-ratio 0.8 --test-ratio 0.2 --recursive True

参数解释:

--list:是否生成lst文件
--prefix:lst文件保存位置
--root:对齐后人脸数据存放位置
--train-ratio/--test-ratio:训练集和测试集的比例

 

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐