Невозможно создать SavedModel из модели keras

Question

Я пытаюсь следовать руководству, найденному здесь , но оно просто сохраняет модель как файл, а не как папку, как описано в руководстве.

Код:

import segmentation_models as sm
from keras.models import load_model
from tensorflow import keras
import tensorflow as tf

path_to_model = "models/unet_70.h5"
loss = sm.losses.binary_focal_dice_loss
model = load_model(path_to_model, custom_objects={'binary_focal_loss_plus_dice_loss': loss})

model.save('model')

Answer 1

Это происходит из-за разницы между Keras api и api для Keras, работающего поверх Tensorflow .

Для достижения желаемого эффекта вы можете изменить импортируемый модуль с

from keras.models import load_model

на

from tensorflow.keras.models import load_model