第三届计图人工智能挑战赛 CGAN
本项目使用Jittor深度学习框架,在数字图片训练集MINIST上训练一个将随机噪声和类别标签映射为数字图片的CGAN模型,生成指定数字序列对应的图片。
安装项目
请参考
https://cg.cs.tsinghua.edu.cn/jittor/download/
完成Jittor的安装。Jittor支持Windows、Linux。mac需要安装虚拟机。
使用项目
执行
python CGAN.py
以运行项目。
程序会将模型保存到generator_last.pkl和 discriminator_last.pkl中。结果会保存到result.png中。同时,会生成如0.png的中间结果文件。
第三届计图人工智能挑战赛 CGAN
本项目使用Jittor深度学习框架,在数字图片训练集MINIST上训练一个将随机噪声和类别标签映射为数字图片的CGAN模型,生成指定数字序列对应的图片。
安装项目
请参考 https://cg.cs.tsinghua.edu.cn/jittor/download/ 完成Jittor的安装。Jittor支持Windows、Linux。mac需要安装虚拟机。
使用项目
执行
以运行项目。
程序会将模型保存到generator_last.pkl和 discriminator_last.pkl中。结果会保存到result.png中。同时,会生成如0.png的中间结果文件。