溫馨提示×

tensorflow怎么加載pb模型

小億
216
2024-04-11 12:30:01

要加載一個已經(jīng)保存好的.pb模型文件，可以使用TensorFlow的tf.saved_model.loader.load函數(shù)。

import tensorflow as tf

# 定義模型路徑
model_path = 'path/to/your/saved_model.pb'

# 加載模型
with tf.Session() as sess:
    loader = tf.saved_model.loader.load(sess, [tf.saved_model.tag_constants.SERVING], model_path)

在加載模型后，你可以通過sess.run()函數(shù)執(zhí)行模型的計(jì)算圖。

0 踩