目录
目录README.md

CGAN_Jittor

本项目为利用计图(jittor)框架编写的Conditional GAN,在项目中,模型将通过训练生成手写体数字照片。

  • 如何安装并配置环境

    首先,利用git将项目仓库克隆到本地

    git clone https://gitlink.org.cn/E6m5t3vaf/CGAN_Jittor.git

    其次,配置python虚拟环境,依次输入以下命令即可

    conda create -n $YOUR_ENV_NAME$ python=3.9
    conda activate $YOUR_ENV_NAME$
    pip install numpy
    python -m pip install jittor
    python -m jittor.test.test_core
    python -m jittor.test.test_example
    python -m jittor.test.test_cudnn_op

    对于计图(jittor)安装,如果有更多问题,请参考计图官方链接!(https://cg.cs.tsinghua.edu.cn/jittor/download/)

  • 如何使用本项目

    进入项目根目录,利用你最喜欢的IDE打开CGAN.py脚本文件或者进入命令行,之后激活虚拟环境,然后输入

    python CGAN.py

    即可运行本项目,你将看到每1000个iter便生成的一张图片以及最终的result.png图片,如果你希望更改生成的数字,可以在CGAN.py文件的第200行处更改number的值

    number = "18513268988" #将number更改为你喜欢的字符串
  • 特别致谢

    本项目的接大多数代码来自清华大学计算机系计算机图形基础教学团队,感谢老师和助教的悉心指导!

  • 协议

    本项目将使用MIT License进行传播

关于

A Jittor implementation of Conditional GAN (CGAN)

14.5 MB
邀请码