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

wendy_ya

暂无认证

  • 1浏览

    0关注

    342博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

基于PyTorch的生成对抗网络入门(3)——利用PyTorch搭建生成对抗网络(GAN)生成彩色图像超详解

wendy_ya 发布时间:2021-11-30 14:41:41 ,浏览量:1

目录
  • 一、案例描述
  • 二、代码详解
    • 2.1 获取数据
    • 2.2 数据集类
    • 2.3 构建判别器
      • 2.3.1 构造函数
      • 2.3.2 测试判别器
    • 2.4 构建生成器
      • 2.4.1 构造函数
      • 2.4.2 测试生成器
    • 2.5 训练GAN
    • 2.6 测试生成器的生成效果

一、案例描述

在上文利用GAN进行手写数字生成的基础上,本文介绍利用生成对抗网络进行彩色人脸图像生成。 本文选取的数据集是CelebA数据集,其官网地址为:http://mmlab.ie.cuhk.edu.hk/projects/CelebA.html。由于该数据集巨大,有202599张图像,因此我们选择前20000张图像进行训练,这20000张图像可以从下方链接获取:

关注
打赏
1659256378
查看更多评论
立即登录/注册

微信扫码登录

0.0391s