将tensorflow保存的模型从tensorflow对象检测api转换为keras模型



是否有办法将TensorFlow对象检测API保存的模型导出为Keras保存的模型格式,或者可以将保存的模型文件转换为Keras保存的模型(.h5)格式?

导入模块:

import tensorflow as tf
from tensorflow.keras.models import load_model

从TensorFlow加载保存的模型:

model = tf.keras.models.load_model('path/to/tf_model')

保存为Keras格式(.h5):

model.save('path/to/keras_model.h5')

应该可以。

最新更新