You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Rosan 7156dcfd51 Update test.py 6 days ago
.ipynb_checkpoints 1.0.0 7 days ago
images 1.0.0 7 days ago
saved_models 1.0.0 7 days ago
CGAN.py Update CGAN.py 6 days ago
README.md Update README.md 7 days ago
d_loss_list.pkl 1.0.0 7 days ago
evaluate.py Update evaluate.py 6 days ago
g_loss_list.pkl 1.0.0 7 days ago
process.py ADD file via upload 7 days ago
result.png 1.0.0 7 days ago
result.zip 1.0.0 7 days ago
run.sh 1.0.0 7 days ago
test.py Update test.py 6 days ago

README.md

Conditional-GAN

本项目使用jittor框架实现,准确说是jittor 1.3.1.53版本。 基于简单MNIST数据集(jittor内置的训练集),完成Conditional-GAN的训练、loss保存。

使用以下命令来获得参数选项

python3.7 CGAN.py -h

使用以下命令训练模型,并保存模型和loss序列:

python3.7 CGAN.py

以下命令可视化loss序列:

python3.7 process.py

生成器测试:

python3.7 test.py

根据提供的一串数字生成图片result.png。 判别器测试:

python3.7 evaluate.py

同样基于jittor提供的MNIST测试集完成。