#tensorflow
Вопрос:
После того, как я сохранил модель tensorflow в формате HDF5, как я могу узнать, с помощью какой версии tensorflow она была построена?
Комментарии:
1. Ответ полезен?
Ответ №1:
Единственное, о чем я знаю при загрузке модели в формате Keras
h5, — это использование model.to_json()
, но вы получите только ту Keras
версию, backend
которая использовалась:
import tensorflow as tf import numpy as np import json model = tf.keras.Sequential([ tf.keras.layers.Input((256, 256, 3)), tf.keras.layers.Conv2D(16, 3, padding='same', activation='relu', strides=1, data_format='channels_last'), tf.keras.layers.MaxPooling2D(), tf.keras.layers.Flatten(), tf.keras.layers.Dense(128, activation='relu'), tf.keras.layers.Dropout(0.1), tf.keras.layers.Dense(5, activation = 'softmax') ]) model.compile(loss='categorical_crossentropy', optimizer='adam') model.save("test.h5", save_format='h5') model = tf.keras.models.load_model('test.h5') meta_data = json.loads(model.to_json()) print('keras_version:', meta_data['keras_version'], 'backend:', meta_data['backend'])
keras_version: 2.7.0 backend: tensorflow
Таким образом, вам придется выяснить, какие Tensorflow
версии совместимы с этой Keras
версией, но это не должно быть слишком сложно.