是否有办法将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')
应该可以。