当前位置：首页 > article >正文

基于python实现的生成对抗网络GAN

article 来源：原创 2023/2/5 16:29:29

项目简介

这篇文章主要介绍了生成对抗网络（Generative Adversarial Network），简称 GAN。

GAN 可以看作是一种可以生成特定分布数据的模型。

2.生成人脸图像

下面的代码是使用 Generator 来生成人脸图像，Generator 已经训练好保存在 pkl 文件中，只需要加载参数即可。由于模型是在多 GPU 的机器上训练的，因此加载参数后需要使用remove_module()函数来修改state_dict中的key。

def remove_module(state_dict_g):# remove module.from collections import OrderedDictnew_state_dict = OrderedDict()for k, v in state_dict_g.items():namekey = k[7:] if k.startswith('module.') else knew_state_dict[namekey] = vreturn new_state_dict

把随机的高斯噪声输入到模型中，就可以得到人脸输出，最后进行可视化。全部代码如下：

import os
import torch.utils.data
import torchvision.transforms as transforms
import torchvision.utils as vutils
import numpy as np
import matplotlib.pyplot as plt
from common_tools i

查看全文

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.dgrt.cn/a/392220.html

如若内容造成侵权/违法违规/事实不符，请联系一条长河网进行投诉反馈，一经查实，立即删除！

基于python实现的生成对抗网络GAN

相关文章：

基于python实现的生成对抗网络GAN

Matlab----绘图以及文件储存

Docker - 12. 容器卷基本概念

PHP 5 MySQLi 函数

论文解读：OMNI-DIMENSIONAL DYNAMIC CONVOLUTION

「Python｜Selenium｜场景案例」如何模拟鼠标单击、双击、右击、长按和拖拽等操作？

【深度学习基础2】生物神经元与感知机

二叉树剑指27

【教程】微服务使用http客户端Feign

Python3 正则表达式

Python与Noah-MP陆面过程模型融合技术及在站点、区域模拟

QMS-云质-质量软件-质量成本如何影响企业利润？

【Python】【进阶篇】三、Pygame的Display显示模块详解

【Note2】macvlan，sol

[QT5.15.2] QtCreator支持中文输入

AutoSar系列之：AutoSar发展

网络安全与防御

【C++从入门到放弃】C/C++内存管理（new和delete的用法详解）

四个上海等保小知识汇总-行云管家

「SQL面试题库」 No_21 当选者