Python3.1 使用卡通头像网络模型生成卡通头像(基于GAN)
生活随笔
收集整理的這篇文章主要介紹了
Python3.1 使用卡通头像网络模型生成卡通头像(基于GAN)
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
使用GAN網(wǎng)絡(luò)生成卡通頭像
提示:當(dāng)前生成方法僅為實(shí)驗(yàn)場(chǎng)景
參考地址
使用GAN網(wǎng)絡(luò)模型生成卡通頭像
- 使用GAN網(wǎng)絡(luò)生成卡通頭像
- 前言
- 一、下載模型
- 二、編寫生成卡通頭像Python代碼
- 1.引入庫
- 2.加載模型
- 3.生成卡通頭像方法
- 總結(jié)
前言
隨著人工智能的不斷發(fā)展,機(jī)器學(xué)習(xí)這門技術(shù)也越來越重要,很多人都開啟了學(xué)習(xí)機(jī)器學(xué)習(xí),本文通過機(jī)器學(xué)習(xí)演示通過訓(xùn)練的模型進(jìn)行卡通頭像的生成。
實(shí)例:
提示:實(shí)驗(yàn)僅供參考
一、下載模型
模型地址:百度網(wǎng)盤模型地址
二、編寫生成卡通頭像Python代碼
1.引入庫
代碼如下:
import tensorflow as tf import time import matplotlib.pyplot as plt import uuid from IPython.display import clear_output2.加載模型
代碼如下:
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() # 獲取當(dāng)前圖像 # 保存卡通頭像到本地目錄,如果沒有目錄需要先創(chuàng)建目錄 f.savefig('D:/space/python/images/{}.png'.format(uuid.uuid1().hex))f.clear() # 釋放內(nèi)存調(diào)用方法代碼如下:
if __name__ == '__main__':generate_cartoon()總結(jié)
以上就是今天要講的內(nèi)容,本文僅僅簡單介紹了對(duì)模型的使用,卡通頭像的生成。
總結(jié)
以上是生活随笔為你收集整理的Python3.1 使用卡通头像网络模型生成卡通头像(基于GAN)的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: PTA乙级1038
- 下一篇: 【渗透测试笔记】之【钓鱼姿势——exe捆