jittor_CGAN

| 第二届计图挑战赛-热身赛

简介

jittor_CGAN使用jittor框架训练Conditional GAN

安装

python3.8 -m pip install jittor export cc_path=”clang++-8”
python3.8 -m jittor.test.test_example export nvcc_path=”/usr/local/cuda/bin/nvcc” python3.8 -m jittor.test.test_cuda
本项目在 1 张 1080TI 上运行，训练时间约为 1 小时。

运行环境

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

安装依赖

执行以下命令安装 python 依赖

pip install -r requirements.txt

包含下面三个依赖 opencv-python==4.1.2.30 numpy==1.18.2 Pillow==6.2.1

训练

python CGAN.py

训练结果

对模型进行训练后，输入限定条件为我的手机号，模型可以输出比较清晰的手机号码图片。