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