
经过600轮epoch训练的DF-GAN生成器CUB-birds文本生成图像模型
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:本项目通过600轮迭代训练开发了DF-GAN生成器,专注于将鸟类(CUB数据集)的描述性文本转化为逼真的图像,实现了高质量的文本到图像合成。
这个是已经训练好的DFGAN模型,用于CUB数据集的生成器部分。该模型经过601轮训练,默认配置文件为bird.yml:
CONFIG_NAME: bird
DATASET_NAME: bird
DATA_DIR: ../data/bird
GPU_ID: 0
WORKERS: 1
B_VALIDATION: True
LOSS FUNCTION: hinge
TREE:
BRANCH_NUM: 1
BASE_SIZE: 256
TRAINING CONFIGURATION:
NF (number of filters): 32,默认为64
BATCH_SIZE: 24
MAX_EPOCHS: 601
NET_G: ../test
TEXT EMBEDDING:
EMBEDDING_DIMENSION: 256
CAPTIONS_PER_IMAGE: 10
预训练的文本编码器路径:
DAMSM_NAME: ../DAMSMencoders/bird/inception/text_encoder200.pth
全部评论 (0)
还没有任何评论哟~


