Jittor 热身赛

| 比赛的训练结果（训练90000次后的结果）如下，result图片包含个人信息未上传

简介

本项目包含了第二届计图挑战赛计图 - 草图生成风景比赛的代码实现。本项目的特点是：采用了 CGAN 方法对 随机采样的噪声 和 给定数字类别 进行处理，取得了 生成指定类别数字的图片 的效果，在评测系统上的评分为 0.9943。

安装

本项目可在1 张 3080ti 上运行，训练时间约为 0.5 小时。

运行环境

• ubuntu 20.04 LTS
• python >= 3.7
• jittor >= 1.3.0

安装依赖

执行以下命令安装 python 依赖

pip install jittor
pip install numpy
pip install Pillow

训练

训练可运行以下命令：

python CGAN.py

训练过程中的生成保存的图像保存于data目录下，可修改CGAN.py第133行修改保存路径

推理

生成图片时可以将修改CGAN.py的第201行修改为想要生成的数字串

预训练模型参数分享在百度网盘中 链接：https://pan.baidu.com/s/11tAqNBfDVfJyB_ukCo7WnQ 提取码：i02t

致谢

此项目基于论文 A Style-Based Generator Architecture for Generative Adversarial Networks 实现，大部分代码参考了 jittor-gan。