python人脸识别训练模型_Python 人脸识别模型训练

--log_base_dir ~/logs/facenet/ 将会把训练日志保存到~/logs/facenet/,可以使用TensorBoard查看这些信息。

--models_base_dir ~/models/facenet/ 最终训练好的模型就保存在~/models/facenet/ 目录下。

--data_dir ~/datasets/casia/casia_maxpy_mtcnnpy_182 训练数据的位置。这里使用之前已经对齐好的CASIA-WebFace数据。

--image_size 160 输入网络的图片尺寸是160x160像素。

--model_def modes.inception_resnet_v1 比较关键的一个参数,它指定了训练的CNN的结构为inception_resnet_v1。项目支持的所有CNN结构在src/models目录下。共支持inception_resnet_v1、inception_resnet_v2、squeezenet三个模型,其中前两个模型较大,最后一个模型较小。如果使用--model_def modes.inception_resnet_v1后,出现内存或显存消耗光的情况,可以将其替换为--model_def modes.squeezenet,来训练一个较小的模型。

--lfw_dir ~/datasets/lfw/lfw_mtcnnpy_160 指定LFW数据集的位置。如果输入这个参数,每次执行完一个epoch,就会在LFW数据集上执行一次测试,并将测试后的正确率写到日志文件中。

--optimizer RMSPROP 指定使用的优化方法。

--learning_rate -1 原意指定学习率,但这里指定率负数,在程序中将忽略这个参数,而使用后面的--learning_rate_schedule_file参数规划学习率。

--max_nrof_epochs 80 表示最多跑80个epoch。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值