DF-GAN实验复现——复现DFGAN详细步骤 及使用MobaXtem实现远程端口到本机端口的转发查看Tensorboard

DF-GAN是南京邮电大学、苏黎世联邦理工学院、武汉大学等学者共同研究开发的一款简单且有效的文本生成图像模型,文章已被CVPR 2022 Oral录用。
DF-GAN最近发布了2022最新版本的代码,代码支持在训练期间自动进行 FID 评估,模型性能和效果很好。
论文地址:https://arxiv.org/abs/2008.05865
代码地址:https://github.com/tobran/DF-GAN
精读与理解:论文精读DF-GAN:一种简单有效的文本生成图像基准模型

MobaXterm 又名 MobaXVT,是一款增强型终端、X 服务器和 Unix 命令集(GNU/ Cygwin)工具箱,是远程操作服务器的优质工具,本篇将使用MobaXterm对DF-GAN进行复现。

一、下载代码及数据

1、使用git clone https://github.com/tobran/DF-GAN.git下载DF-GAN最新代码,下载后结构如下:
在这里插入图片描述
2、下载鸟数据集的元处理数据:https://drive.google.com/file/d/1I6ybkR7L64K8hZOraEZDuHh0cCJw5OUj/view?usp=sharing
打不开可以通过CSDN下载:DF-GAN-CUB-元处理数据包
元数据包下载完毕后如下图所示:
在这里插入图片描述
将元数据包的文件放入到DF-GAN/data/birds目录下,如图所示:
在这里插入图片描述

3、下载CUB-Bird鸟图像数据集,下载链接:
http://www.vision.caltech.edu/visipedia/CUB-200-2011.html,并将其保存在并将其保存在data/birds/中,最终如图所示:
在这里插入图片描述

二、配置环境

创建虚拟环境,然后进入DF-GAN目录 ,使用命令行:pip install -r requirements.txt安装必备的环境,也可以手动安装(不推荐),环境要求如下:
在这里插入图片描述

三、训练

对于鸟类数据集,进入DF-GAN/code目录,命令行输入:bash scripts/train.sh ./cfg/bird.yml,当出现类似如下情况则说明训练过程正常运行。
在这里插入图片描述
如果命令行输入的是以上格式,需要MobaXtem一直在线连接,给输出提供传输信道,本地关机后程序就会终止,那么我们可以使用nohup命令将输出放入output.txt:

nohup bash scripts/train.sh ./cfg/bird.yml >output.txt 2>&1 &

但是这种方式,在运行一段时间后,会把output.txt写的非常非常大,占用几个G的空间。既然DF-GAN提供了Tensorboard,我们干脆不写进output.txt了,直接使用Tensorboard监控就行,于是我们把输出写进/dev/null,(dev/null就是一个黑洞,所有写进这个目录的文件都会凭空消失)命令如下:

 nohup bash scripts/train.sh ./cfg/bird.yml >/dev/null 2>&1 &

四、使用Tensorboard监控数据

DF-GAN最新版本的代码支持在训练期间自动进行 FID 评估,结果存储在 ./logs 下的 TensorBoard 文件中,在命令行输入:tensorboard --logdir=./code/logs/bird/train --port 8166,会显示如下:
在这里插入图片描述

然后利用MobaXterm建立ssh隧道(使用本机复现实验的话不需要看后面的步骤,直接访问http://127.0.0.1:8166/),实现远程端口到本机端口的转发,打开tunneling:
在这里插入图片描述
新建SSH通道:
在这里插入图片描述
填入端口和用户名
在这里插入图片描述
点击start开始转发:
在这里插入图片描述
然后就可以在本机访问http://127.0.0.1:8166/,查看tensorboard了,
在这里插入图片描述

DF-GAN是一种用于图像生成的网络模型。这个网络模型的目标是通过学习真实图像的分布来生成类似真实图像的虚拟图像DF-GAN包含两个关键组件:生成器网络和判别器网络。生成器网络负责接收一个随机噪声向量,并生成一个与真实图像相似的图像。这个网络通常包含多个隐藏层,每个隐藏层都使用一种特殊的层结构(例如卷积层、批次规范化层等),以帮助生成逼真的图像生成器网络通过训练过程不断优化自己的参数,以使生成图像越来越接近真实图像。 判别器网络的任务是对生成生成图像和真实图像进行区分。它接收一个图像作为输入,然后根据其判断其是否是真实图像。判别器网络也是通过训练过程不断调整参数,以提高区分真实和虚拟图像的准确性。 为了提高DF-GAN的性能,研究人员还引入了梯度惩罚(gradient penalty)技术。这个技术的目的是约束生成器和判别器的梯度大小,以确保生成器和判别器之间的平衡。通过限制梯度的大小,可以避免生成器和判别器网络之间的不稳定性。 总的来说,DF-GAN是一种用于图像生成深度学习模型。它通过生成器和判别器网络的协同训练,学习真实图像的特征分布,并生成与真实图像相似的虚拟图像。通过使用梯度惩罚技术,DF-GAN可以提高生成器和判别器之间的平衡,从而提高生成图像的质量。
评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

中杯可乐多加冰

请我喝杯可乐吧,我会多加冰!

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值