上传者: 42109125
|
上传时间: 2021-10-19 17:13:14
|
文件大小: 26KB
|
文件类型: -
达甘
DAGAN的实现:数据增强生成对抗网络
介绍
这是描述的DAGAN的实现。 该实现为Omniglot和VGG-Face数据集提供了数据加载器,模型构建器,模型训练器和合成数据生成器。
安装
要使用DAGAN存储库,您必须首先安装项目依赖项。 这可以通过从使用python 3安装miniconda3并运行来完成:
pip install -r requirements.txt
数据集
可以在以numpy格式获取Omniglot和VGG-Face数据集。 然后将它们放置在datasets文件夹中。
训练DAGAN
在下载数据集并安装依赖项之后,可以通过运行以下命令来训练DAGAN:
python train_omniglot_dagan.py --batch_size 32 --generator_inner_layers 3 --discriminator_inner_laye