目录
目录README.md

生成手写数字-CGAN

用户随机ID为542682542720

主要结果

简介

本项目包含了第二届计图挑战赛计图 - 生成手写数字比赛的代码实现。本项目的特点是:采用了Conditional GAN方法在数字图片数据集MNIST上进行训练, 建立一个将随机噪声和类别标签映射为数字图片的模型。最终,对用户随机ID进行测试,得到对应的手写数字的图像。

安装

本项目可在 1 张 4070 上运行,训练时间约为 10 分钟。

运行环境

  • windows 10
  • python == 3.10
  • jittor == 1.3.8

安装依赖

执行以下命令安装 python 依赖

conda create --name <envName> --file requirements.txt

训练

单卡训练可运行以下命令:

python CGAN.py

推理

模型会自动生成最终结果,并保存到result.png中。

致谢

此项目基于论文 Conditional Generative Adversarial Nets 实现,部分代码参考了 jittor-gan

关于

This project task is the generation of handwritten numbers, and part of the code references jittor-gan

41.0 KB
邀请码
    Gitlink(确实开源)
  • 加入我们
  • 官网邮箱:gitlink@ccf.org.cn
  • QQ群
  • QQ群
  • 公众号
  • 公众号

©Copyright 2023 CCF 开源发展委员会
Powered by Trustie& IntelliDE 京ICP备13000930号