探究CVAE(条件自编码) Condition GAN (条件GAN) 和 VAE-GAN模型之间的区别之程序入口函数
2019-03-20 16:29
246 查看
版权声明:版权归世界上所有无产阶级所有 https://blog.csdn.net/qq_41776781/article/details/88692238
三个程序的入口地址: main函数()
程序运行的命令是: CUDA_VISIBLE_DEVICES=0 python main --gan_type F_GAN
[code]import os from CGAN import CGAN from CVAE import CVAE from F_GAN import F_GAN from utils import check_folder import tensorflow as tf import argparse """parsing and configuration""" def parse_args(): desc = "Tensorflow implementation of GAN collections" parser = argparse.ArgumentParser(description=desc) parser.add_argument('--gan_type', type=str, default='F_GAN', choices=['CGAN', 'CVAE', 'F_GAN'], help='The type of GAN', required=True) parser.add_argument('--dataset', type=str, default='fashion-mnist', choices=['mnist', 'fashion-mnist', 'celebA'], help='The name of dataset') parser.add_argument('--epoch', type=int, default=500, help='The number of epochs to run') parser.add_argument('--batch_size', type=int, default=64, help='The size of batch') parser.add_argument('--z_dim', type=int, default=100, help='Dimension of noise vector') parser.add_argument('--result_dir', type=str, default='results', help='Directory name to save the generated images') return check_args(parser.parse_args()) def check_args(args): # if not exited then create the result file check_folder(args.result_dir) return args """main""" def main(): # parse arguments args = parse_args() if args is None: exit() models = ['CGAN', 'CVAE', 'F_GAN'] with tf.Session(config=tf.ConfigProto(allow_soft_placement=True)) as sess: if args.gan_type in models: gan = model(sess, epoch=args.epoch, batch_size=args.batch_size, z_dim=args.z_dim, dataset_name=args.dataset, result_dir=args.result_dir,) gan.build_model() gan.train() if __name__ == '__main__': main()
相关文章推荐
- CVAE(条件自编码) Condition GAN (条件GAN) 和 VAE-GAN模型之间的区别之CVAE
- CVAE(条件自编码) Condition GAN (条件GAN) 和 VAE-GAN模型之间的区别值 VAE-GAN
- CVAE(条件自编码) Condition GAN (条件GAN) 和 VAE-GAN模型之间的区别之Condition GANs
- CVAE(条件自编码) Condition GAN (条件GAN) 和 VAE-GAN模型之间的区别
- 在后端是PHP程序的情况下,保持前端Javascript和PHP之间传值的统一编码可以使用以下函数进行处理:
- 三个可以对字符串编码的函数escape()、encodeURI()、encodeURIComponent()之间的区别详解
- 设计程序,单击【随机数】按钮,使用Math对象的random函数产生一个0-100之间(含0-100)的随机整数,并在对话框中显示,如下图。单击【计算】按钮,计算该随机数的平方、平方根和自然对数,保留两位小数,并在对话框中显示,如下图。
- UTF-8和GBK编码之间的区别(页面编码、数据库编码区别)以及在实际项目中的应用
- Windows To Go 和典型的 Windows 安装程序之间的区别
- MFC中消息循环处理的几个函数之间的区别
- 多态-成员变量-成员函数-静态函数 之间的注意点与区别
- Java中static方法、程序入口函数main方法的继承问题
- JavaScript中各种编码解码函数的区别和注意事项
- 裸机程序常识1 _start程序入口标号 -Ttext链接地址作用,如何使用uboot中的函数
- 深入了解scanf() getchar()和gets()等函数之间的区别
- VAE、GAN、Info-GAN:全解深度学习三大生成模型
- 父子类之间,成员函数重写、重载以及重定义的区别
- 数据库中存储过程与自定义函数之间的区别
- 程序入口函数和glibc及C++全局构造和析构
- ansi,unicode,gb2312,utf-8编码之间的区别