471eb9eb52267041d68d1ffad116cd29330fca30,stylize_image.py,,,#,10

Before Change

    args = parser.parse_args()

    logger.info("Loading model from %s" % args.model_checkpoint)
    transfer_net = models.StyleTransferNetwork.build(
        args.img_height, args.img_width
    )
    transfer_net.load_weights(args.model_checkpoint)

    inputs = [transfer_net.input, keras.backend.learning_phase()]
    outputs = [transfer_net.output]

After Change

    args = parser.parse_args()

    logger.info("Loading model from %s" % args.model_checkpoint)
    custom_objects = {
        "InstanceNormalization":
            keras_contrib.layers.normalization.InstanceNormalization,
        "DeprocessStylizedImage": layers.DeprocessStylizedImage
    }
    transfer_net = keras.models.load_model(
        args.model_checkpoint,
        custom_objects=custom_objects
    )

    image_size = transfer_net.input_shape[1:3]

    inputs = [transfer_net.input, keras.backend.learning_phase()]
    outputs = [transfer_net.output]

    transfer_style = keras.backend.function(inputs, outputs)

    input_image = utils.load_image(
        args.input_image,
        image_size[0],
        image_size[1],
        expand_dims=True
    )
    output_image = transfer_style([input_image, 1])[0]

Instances