目录
目录README.md

CGAN_jittor

Jittor License: MIT

A Jittor implementation of Conditional GAN (CGAN) for the 3rd Jittor AI Challenge.

简介 (Introduction)

本项目基于清华大学开源的深度学习框架 Jittor,实现了一个条件生成对抗网络 (Conditional GAN)。该模型在经典的 MNIST 数据集上进行训练,能够根据用户提供的特定数字标签(如 “2312123”),生成对应的高质量手-写数字图像。

这个项目是计图人工智能挑战赛热身赛的官方赛题实现。

环境依赖 (Requirements)

  • Python 3.9+
  • Jittor (本项目测试版本为 1.3.7.15)
  • Numpy
  • Pillow (PIL for image processing)
  • Matplotlib (用于绘制损失曲线)

如何运行 (Usage)

  1. 克隆或下载本项目:

    git clone https://gitlink.org.cn/lyyoo/CGAN_jittor.git
    cd CGAN_jittor
  2. 安装依赖 (推荐在 Conda 环境中):

    pip install jittor==1.3.7.15 numpy Pillow matplotlib
  3. 开始训练和生成:

    python CGAN.py

    训练过程大约需要 30 分钟。训练完成后,最终生成的、符合赛题要求的图片 result.png 和损失曲线图 loss_curve.png 会保存在项目根目录下。

结果展示 (Result)

最终生成图片 (Final Generated Image)

根据数字序列 “2312123” 生成的图片如下:

最终结果

损失函数曲线 (Loss Curve)

训练过程中的生成器 (Generator) 与判别器 (Discriminator) 的损失变化曲线如下:

Loss Curve
关于

A Jittor implementation of Conditional GAN (CGAN).

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

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