附上代码连接:align_dataset_mtcnn.py
该代码主要是用于人脸对齐和人脸图像缩略图存储用的。人脸对齐主要是将人的鼻子,眼睛,嘴巴等特征标识出来。人脸对齐之前需要做的工作是人脸检测,之后人脸对齐,接着才可以进行人脸识别或者人脸验证等。
该脚本主要由两个函数构成,main函数和parse_arguments参数解析.parse_arguments函数比较简单,具体使用可参考python内置模块argparse的使用方法。主要是提供了两个必需参数input_dir,output_dir,以及若干个可选参数image_size,margin等。通过设置这些参数,使得我们无需更改脚本,即可在运行程序的时候自己设置参数值。使用方式为 python align_dataset_mtcnn.py your_input_dir_path your_output_dir_path –image_size 200…。
#main函数是此代码的主函数。
sleep(random.random())
output_dir = os.path.expanduser(args.output_dir)
if not os.path.exists(output_dir):
os.makedirs(output_dir)
上述代码含义为随机挂起代码若干秒,parse_arguments返回的参数中获取输出目录。检查该目录是否存在,如果不存在,则建立该目录。
src_path,_ = os.path.split(os.path.realpath(__file__))
获取当前文件所在目录;facenet.store_re