目录
目录README.md

Jittor计图挑战热身赛热身赛(基于CGAN的手写数字生成)

简介

本项目为第二届计图人工智能挑战赛-热身赛的代码实现。项目将会加载数字图片数据集 MNIST,训练一个将随机噪声和类别标签映射为数字图片的CGAN模型,并生成注册时绑定的手机号。

安装

本项目将在一张1080Ti的显卡上进行训练和预测。

运行环境

  • windows 10
  • python >= 3.8
  • jittor >= 1.3.0
  • (可选)cuda >= 11.0

安装依赖

在cmd中执行以下命令安装项目必须的Python依赖。

pip install jittor
pip install numpy

训练

在CGAN.py的同级目录下执行以下命令训练并得到程序中赋值给number字符串的手机号码手写数字:

python CGAN.py

训练结束后,默认将训练好的模型参数保存到代码的同级目录下,生成器和判别器的文件名分别为: generator_last.pkl discriminator_last.pkl

致谢

此项目基于论文 A Style-Based Generator Architecture for Generative Adversarial Networks 中提出的原理实现,部分代码参考了jittor-gan

关于

A conditional GAN by Jittor to generate specific handwritten digits

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

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