kerasの学習済みモデルをpbファイルで出力する - Qiita

import tensorflow as tf from tensorflow.python.framework.convert_to_constants import convert_variables_to_constants_v2 model = tf.keras.applications.vgg16.VGG16(weights='imagenet') model.save("tf_vgg16", save_format='tf') new_model = tf.keras.models.load_model('tf_vgg16') full_model = tf.function(lambda inputs: new_model(inputs, training=False)) # カスタムレイヤーがある場合は、例えば以下のように書く # new_model = tf.keras.