Как узнать версию Tensorflow, используемую в модели HDF5

#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 версией, но это не должно быть слишком сложно.