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
训练结果
对模型进行训练后,输入限定条件为我的手机号,模型可以输出比较清晰的手机号码图片。
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 小时。
运行环境
安装依赖
执行以下命令安装 python 依赖
包含下面三个依赖 opencv-python==4.1.2.30 numpy==1.18.2 Pillow==6.2.1
训练
python CGAN.py
训练结果
对模型进行训练后,输入限定条件为我的手机号,模型可以输出比较清晰的手机号码图片。