谷歌facenet项目代码解读（一）--align_dataset_mtcnn.py

最新推荐文章于 2023-07-14 14:33:45 发布

原创

最新推荐文章于 2023-07-14 14:33:45 发布 · 2.4k 阅读

3 ·

CC 4.0 BY-SA版权

本文详细解读了`align_dataset_mtcnn.py`代码，该代码用于人脸对齐和图像缩略图生成。首先介绍人脸检测，然后进行对齐操作，为后续的人脸识别或验证做准备。脚本包含main函数和parse_arguments参数解析，通过argparse模块提供可定制的参数。文章结尾预告了下节将深入探讨人脸对齐技术。

附上代码连接：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_revision_info(src_path, output_dir, ' '.join(sys.argv)) 将git修订信息存储到日志目录文档中。dataset = facenet.get_da

最低0.47元/天解锁文章