ADD file via upload
基于CGAN和MNIST数据集的数字图片生成模型
本项目包含了第二届计图挑战赛计图 - 数字图片生成热身赛的代码实现。
ubuntu 18.04 LTS python >= 3.7 jittor >= 1.3.0
在终端执行以下命令来安装所需要的依赖包： pip install -r requirements.txt jittor的安装方法请参考 https://cg.cs.tsinghua.edu.cn/jittor/download/
训练可运行以下命令，数据集将被自动下载，且开始训练和生成设定好的数字图片：
python CGAN.py –batch_size 32
基于Jittor框架下开发的图片生成任务，通过在MNIST上训练CGAN模型，来生成特定数字的图像
版权所有：中国计算机学会技术支持：开源发展技术委员会 京ICP备13000930号-9 京公网安备 11010802032778号
简介
基于CGAN和MNIST数据集的数字图片生成模型
本项目包含了第二届计图挑战赛计图 - 数字图片生成热身赛的代码实现。
运行环境
ubuntu 18.04 LTS python >= 3.7 jittor >= 1.3.0
安装依赖
在终端执行以下命令来安装所需要的依赖包： pip install -r requirements.txt jittor的安装方法请参考 https://cg.cs.tsinghua.edu.cn/jittor/download/
训练和推理
训练可运行以下命令，数据集将被自动下载，且开始训练和生成设定好的数字图片：
python CGAN.py –batch_size 32