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模型，来生成特定数字的图像
©Copyright 2023 CCF 开源发展委员会 Powered by Trustie& IntelliDE 京ICP备13000930号
简介
基于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