Я хотел бы удалить выпадающие слои предварительно обученной модели (unet с основой efficien tnet), потому что она использует фиксированный отсев, который не поддерживается coremltools при попытке преобразования в модель ios.
Я пробовал это решение , но я получаю
ValueError: A merge layer should be called on a list of inputs.
Код:
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})
layers = [l for l in model.layers]
x = layers[0](layers[0].input)
for i in range(1, 68):
layers[i].trainable = False
x = layers[i](x)
for i in range(69, len(layers)):
layers[i].trainable = False
x = layers[i](x)