您当前的位置: 首页 >  网络

wendy_ya

暂无认证

  • 1浏览

    0关注

    342博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

基于PyTorch的生成对抗网络实战(1)——利用Pytorch搭建GAN生成手写数字

wendy_ya 发布时间:2021-10-12 15:56:11 ,浏览量:1

目录
    • 一、生成器
    • 二、判别器
    • 三、迭代训练过程
    • 四、结果演示
    • 五、完整代码

一、生成器

生成器的目标是输入一个固定维度的随机噪声,生成手写数字图像,因此它的输入是一个长度为N=100的一维的向量,输出一个28,28,1维的图像。 输入一个100维的0~1之间的高斯分布,然后通过第一层线性变换将其映射到128维,然后通过LeakyReLU激活函数,接着进行一个线性变换将其变到256维,再经过一个LeakyReLU激活函数,然后经过线性变换将其变成512维,再经过一个LeakyReLU激活函数,然后经过线性变换将其变成1024维,最后经过线性变换将其变成784维(即28281维),最后经过Tanh激活函数是希望生成的假的图片数据分布能够在-1~1之间。 生成器类代码:

class Generator(nn.Module)
关注
打赏
1659256378
查看更多评论
立即登录/注册

微信扫码登录

0.0397s