Python3.1 使用卡通头像网络模型生成卡通头像(基于GAN)

使用GAN网络生成卡通头像

提示：当前生成方法仅为实验场景
参考地址

前言

随着人工智能的不断发展，机器学习这门技术也越来越重要，很多人都开启了学习机器学习，本文通过机器学习演示通过训练的模型进行卡通头像的生成。
实例：

提示：实验仅供参考

一、下载模型

模型地址：百度网盘模型地址

二、编写生成卡通头像Python代码

1.引入库

代码如下：

import tensorflow as tf
import time
import matplotlib.pyplot as plt
import uuid
from IPython.display import clear_output

2.加载模型

代码如下：

cartoon_model = tf.keras.models.load_model('./cartoon_model/', compile=False)

3.生成卡通头像方法

def generate_cartoon():
    """
    通过模型生成卡通头像
    """
    noise_seed = tf.random.normal([16, 128])
    predictions = cartoon_model(noise_seed, training=False)
    plt.figure(figsize=(32, 32))
    for i in range(predictions.shape[0]):
        # 生成的图片中放多少张卡通头像，四行四列
        plt.subplot(4, 4, i + 1)
        img = (predictions[i].numpy() * 255).astype('int')
        plt.imshow(img)
        plt.axis('off')
    # plt.show()
    f = plt.gcf()  # 获取当前图像
# 保存卡通头像到本地目录，如果没有目录需要先创建目录    f.savefig('D:/space/python/images/{}.png'.format(uuid.uuid1().hex))
    f.clear()  # 释放内存

调用方法代码如下：

if __name__ == '__main__':
    generate_cartoon()
    

总结

以上就是今天要讲的内容，本文仅仅简单介绍了对模型的使用，卡通头像的生成。