双十二人像变换推荐

import tensorflow as tf
import tensorflow_hub as hub
import numpy as np
import PIL.Image

def load_image(image_path):
    img = tf.io.read_file(image_path)
    img = tf.image.decode_image(img, channels=3)
    img = tf.image.convert_image_dtype(img, tf.float32)
    img = img[tf.newaxis, :]
    return img

def save_image(image, filename):
    image = np.squeeze(image)
    image = np.clip(image * 255, 0, 255).astype(np.uint8)
    img = PIL.Image.fromarray(image)
    img.save(filename)

content_image = load_image("path_to_content_image.jpg")
style_image = load_image("path_to_style_image.jpg")

hub_module = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2')
stylized_image = hub_module(tf.constant(content_image), tf.constant(style_image))[0]

save_image(stylized_image, "output_stylized_image.jpg")