微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

如何解决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 举报,一经查实,本站将立刻删除。