文件名称:Conditional_GAN_Mnist
文件大小:27.39MB
文件格式:ZIP
更新时间:2024-04-02 09:36:29
Python
有条件的_GAN_Mnist 抽象的 该项目是我们团队在Kaist大学进行的CS492“机器学习计算机视觉”课程的最终项目。 在该项目中,我们为Mnist数据集实现了条件GAN模型。 安装 keras == 2.4.3 张量流== 2.4.0 火车 要进行训练,只需运行文件cgan.py。 在训练过程中经过一定时间后,模型的权重将保存到文件夹“ saved_model_weights”,模型生成的图像将保存到文件夹“ images”。 训练完成后,训练历史记录将保存到文件夹“ plot_history_training”。 测试 对于演示,我曾经训练过一次模型,并将生成的图像以及模型权重上传到两个文件夹“ images / version1”和“ saved_model_weights / version1”中。 要测试经过训练的模型的性能,请运行文件test_cgan.py。 经