facenet加载pretrained_model报错问题解决

最新推荐文章于 2024-03-08 22:03:50 发布

白小心

最新推荐文章于 2024-03-08 22:03:50 发布

阅读量1.3k

点赞数 1

本文链接：https://blog.csdn.net/qq_26794857/article/details/88095473

版权

在使用facenet训练模型过程中遇到加载预训练模型报错。原因是调用tf.train.Saver时，参数错误，未指向正确的模型文件。解决办法包括：1. 输入正确的模型文件路径；2. 修改`src/train_softmax.py`文件，指定模型文件夹路径。

摘要由CSDN通过智能技术生成

facenet加载pretrained_model报错问题解决

问题

facenet github地址：https://github.com/davidsandberg/facenet
使用facenet自己训练模型中断后，继续运行程序：

// 继续训练
python3 src/train_softmax.py --pretrained_models models/20190303-192600/ --其他参数不写了

报错。
注：models/20190303-192600是上次自己训练保存模型的文件夹

原因

Facenet程序调用了tf.train.Saver类，加载预训练模型时使用了Saver.restore(sess, pretrained_model)方法，该方法传入的第二个变量应该为models/20190303-192600/model-20190303-192600.ckpt-6的字符串。
对比models/20190303-192600文件夹下的内容，发现正确输入的参数并不指向预训练模型文件夹下任何一个文件。
models/20190303-192600文件夹下的内容：