migrate.py

import keras.backend as K
import numpy as np

from model import build_encoder_decoder
from vgg16 import vgg16_model


def migrate_model(new_model):
    old_model = vgg16_model(224, 224, 3)
    # print(old_model.summary())
    old_layers = [l for l in old_model.layers]
    new_layers = [l for l in new_model.layers]

    old_conv1_1 = old_model.get_layer('conv1_1')
    old_weights = old_conv1_1.get_weights()[0]
    old_biases = old_conv1_1.get_weights()[1]
    new_weights = np.zeros((3, 3, 3, 64), dtype=np.float32)
    new_weights[:, :, 0:3, :] = old_weights
    new_conv1_1 = new_model.get_layer('conv1_1')
    new_conv1_1.set_weights([new_weights, old_biases])

    for i in range(2, 31):
        old_layer = old_layers[i]
        new_layer = new_layers[i + 1]
        new_layer.set_weights(old_layer.get_weights())

    del old_model


if __name__ == '__main__':
    model = build_encoder_decoder()
    migrate_model(model)
    print(model.summary())
    model.save_weights('models/model_weights.h5')

    K.clear_session()