Как преобразовать сеанс Tensorflow в объект модели keras?

Question

Предположим, у меня есть предварительно обученная модель, сохраненная в контрольной точке Tensorflow.Я хотел бы преобразовать его в модель Keras.Я могу загрузить контрольную точку в сеанс TF, но я застрял там.

Answer 1

Я думаю, что невозможно создать модель Keras с использованием контрольной точки TF, но вы можете скопировать ее веса в уже созданную модель Keras.

Оформить заказ. https://github.com/yuyang-huang/keras-inception-resnet-v2

extract_weights.py предназначен для сохранения весов TF в массиве numpy, а load_weights.py для загрузки файла npy в модель Keras.

Для справки, вот как я это реализую https://github.com/DableUTeeF/keras-efficientnet/tree/master/keras_efficientnet.