相关疑难解决方法(0)

Tensorflow (.pb) 格式到 Keras (.h5)

我正在尝试将 Tensorflow (.pb) 格式的模型转换为 Keras (.h5) 格式以查看事后注意可视化。我试过下面的代码。

file_pb = "/test.pb"
file_h5 = "/test.h5"
loaded_model = tf.keras.models.load_model(file_pb)
tf.keras.models.save_keras_model(loaded_model, file_h5)
loaded_model_from_h5 = tf.keras.models.load_model(file_h5)
Run Code Online (Sandbox Code Playgroud)

谁能帮我这个?这甚至可能吗?

machine-learning python-3.x deep-learning keras tensorflow

4
推荐指数
1
解决办法
8670
查看次数