tf.keras により、モデルの重みを HDF5 と TensorFlow の 2 種類のフォーマットで保存できます。このツールでサポートされるのは、現在のところ TensorFlow フォーマットのみです。モデルの重みが HDF5 フォーマットで保存されている場合は、次の手順に進む前に TensorFlow フォーマットに変換してください。
import tensorflow as tf
tf.keras.backend.set_learning_phase(0)
model = tf.keras.applications.ResNet50(weights="imagenet",
include_top=True,
input_tensor=None,
input_shape=None,
pooling=None,
classes=1000)
model.save_weights("model.ckpt", save_format='tf')
変換されたチェックポイント ファイルは次のようになります。
model.ckpt.data-00000-of-00001
model.ckpt.data-00001-of-00002
model.ckpt.index