Удалить промежуточные слои функциональной модели керас - PullRequest
0 голосов
/ 18 апреля 2020

Я хотел бы удалить выпадающие слои предварительно обученной модели (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)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...