上传者: 42106299
|
上传时间: 2022-03-31 20:15:21
|
文件大小: 138KB
|
文件类型: -
适用于小型和高分辨率图像集的快速稳定的GAN-pytorch
该文件的正式pytorch实施“走向更快,更稳定的GAN训练,以实现高保真的少量拍摄图像合成”,可在找到该文件。
0.数据
本文中使用的数据集可以在上找到。
在对20多个数据集进行测试后,每个数据集的图像少于100个,该GAN会收敛到其中的80%。 对于该GAN可以收敛的数据集,我仍然无法总结出明显的“良好属性”模式,请随时尝试使用您自己的数据集。
1.说明
该代码的结构如下:
models.py:所有模型的结构定义。
operation.py:培训过程中的辅助功能和数据加载方法。
train.py:代码的主要条目,执行此文件以训练模型,中间结果和检查点将自动定期保存到文件夹“ train_results”中。
eval.py:将图像从受过训练的生成器生成到文件夹中,该文件夹可用于计算FID分数。
基准测试:我们用