davidsandberg/facenet 如何在 TF2.x 中加载模型

问题描述

我刚开始使用 TF2，在网上找不到任何相关信息。请帮助我的无知

我从以下链接位置 https://github.com/davidsandberg/facenet

下载了 https://drive.google.com/open?id=1EXPBSXwTaqrSC0OhUdXNmKSh9qJUQ55- 的预训练模型

我下载了 4 个文件（扩展名为 .pb、.Meta、.index 和权重的文件）。但我不知道接下来要做什么。

我尝试了代码片段

from keras.models import load_model
import tensorflow.compat.v1 as v1
#tf.disable_v2_behavior()
print(v1.__version__)
def load_pb(path_to_pb):
    with v1.gfile.GFile(path_to_pb,"rb") as f:
        graph_def = v1.GraphDef()
        graph_def.ParseFromString(f.read())
    with v1.Graph().as_default() as graph:
        v1.import_graph_def(graph_def,name='')
        return graph
model = load_pb('saved_model.pb')

它运行时没有错误，但我不知道下一步该怎么做，因为它是 Tensorflow 1 和图表。

我试图从链接中理解图表，例如： https://www.tensorflow.org/api_docs/python/tf/Graph https://www.tensorflow.org/api_docs/python/tf/saved_model/load

但我不是很懂

我也试过 loaded = tf.saved_model.load("./","saved_model.pb") 没有取得多大成功。

有人可以给我一些指导吗？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

face-recognition tensorflow2.x