Python3.1 使用卡通头像网络模型生成卡通头像(基于GAN)_python人物头像的卡通化算法设计与实现

使用GAN网络生成卡通头像

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

---

---

前言

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

---

提示：实验仅供参考

一、下载模型

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

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

1.引入库

代码如下：

import tensorflow as tf
import time
import matplotlib.pyplot as plt
import uuid
from IPython.display import clear_output
1
2
3
4
5

2.加载模型

代码如下：

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

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()  # 释放内存

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
'
运行
运行

调用方法代码如下：

if __name__ == '__main__':
    generate_cartoon()
    
1
2
3

---

总结

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