30、图像风格迁移技术详解与实践

图像风格迁移技术详解与实践

1. 快速风格迁移

1.1 加载预训练模块

要将新风格应用到目标图像上，需要从 tfhub 加载预训练模块。使用以下语句：

import tensorflow_hub as hub
hub_module = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2')

这里的模块名为 arbitrary-image-stylization-v1-256/2 。

1.2 执行风格迁移

模块加载完成后，将两个张量作为输入传递给模块进行转换：

import tensorflow as tf
outputs = hub_module(tf.constant(target_image), tf.constant(style_image))
stylized_image = outputs[0]

outputs 是一个形状为 (1, 300, 400, 3) 的张量，这是转换后图像的数据。注意， 300x400 是原始 1600x1200 图像的缩小版本。

1.3 显示输出图像

要